diff --git a/tests/test_docs.py b/tests/test_docs.py index 837f7e865c..ca83c18f58 100644 --- a/tests/test_docs.py +++ b/tests/test_docs.py @@ -1,4 +1,4 @@ -from invokeai.app.services.config.config_default import InvokeAIAppConfig +from invokeai.app.services.config.config_default import InvokeAIAppConfig, generate_config_docstrings def test_app_config_docstrings_are_current(): @@ -9,7 +9,7 @@ def test_app_config_docstrings_are_current(): assert InvokeAIAppConfig.__doc__ is not None - generated_docstring = InvokeAIAppConfig.generate_docstrings() + generated_docstring = generate_config_docstrings() formatted_dunder_docstring = f' """{InvokeAIAppConfig.__doc__.strip()}\n """'