fix model schema file type (#43)

kenho811 · Oct 15, 2023 · e6844e7 · e6844e7
2 parents 7c21adf + 9489dd0
commit e6844e7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/dump_model_schemas.py b/scripts/dump_model_schemas.py
@@ -11,7 +11,8 @@ def main(output_folder_path: Path):
     models: List[BaseModel] = sum([entry.models for entry in ProcessorRegistry().registry], [])
     for model in models:
         print('============================')
-        file_path = output_folder_path.joinpath(model.model_name)
+        file_name = f"{model.model_name}.json"
+        file_path = output_folder_path.joinpath(file_name)
         schema: Dict = model.model_json_schema()
         with open(file_path, 'w') as schema_file:
             schema_str: str = json.dumps(schema, indent=4)