fix: fix the write_all* scripts

Haidra-Org · Jul 11, 2023 · 91fc0f6 · 91fc0f6
1 parent 87d6779
commit 91fc0f6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/horde_sdk/scripts/write_all_payload_examples.py b/src/horde_sdk/scripts/write_all_payload_examples.py
@@ -5,7 +5,9 @@
 
 
 def main(*, test_data_path: Path | None = None):
-    ai_horde_swagger_doc = SwaggerParser(get_ai_horde_swagger_url()).get_swagger_doc()
+    ai_horde_swagger_doc = SwaggerParser(
+        swagger_doc_url=get_ai_horde_swagger_url(),
+    ).get_swagger_doc()
     if not test_data_path:
         test_data_path = (
             Path(__file__).parent.parent.parent.parent / "tests" / "test_data" / "ai_horde_api" / "example_payloads"

diff --git a/src/horde_sdk/scripts/write_all_response_examples.py b/src/horde_sdk/scripts/write_all_response_examples.py
@@ -5,7 +5,9 @@
 
 
 def main(*, test_data_path: Path | None = None):
-    ai_horde_swagger_doc = SwaggerParser(get_ai_horde_swagger_url()).get_swagger_doc()
+    ai_horde_swagger_doc = SwaggerParser(
+        swagger_doc_url=get_ai_horde_swagger_url(),
+    ).get_swagger_doc()
     if not test_data_path:
         test_data_path = (
             Path(__file__).parent.parent.parent.parent / "tests" / "test_data" / "ai_horde_api" / "example_responses"