fix: set new type hint for other requirements fields

Also updates the automatically generated schema files
Haidra-Org · Sep 30, 2024 · a3e2377 · a3e2377
1 parent 8bd5607
commit a3e2377
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 3 deletions.
diff --git a/horde_model_reference/legacy/classes/raw_legacy_model_database_records.py b/horde_model_reference/legacy/classes/raw_legacy_model_database_records.py
@@ -59,7 +59,7 @@ class RawLegacy_StableDiffusion_ModelRecord(BaseModel):
     homepage: str | None = None
     nsfw: bool
     download_all: bool
-    requirements: dict[str, int | float | str | list[int] | list[str] | bool] | None = None
+    requirements: dict[str, int | float | str | list[int] | list[float] | list[str] | bool] | None = None
     config: Mapping[str, list[RawLegacy_FileRecord | RawLegacy_DownloadRecord]]
     available: bool | None = None
     features_not_supported: list[FEATURE_SUPPORTED] | None = None

diff --git a/horde_model_reference/legacy/classes/staging_model_database_records.py b/horde_model_reference/legacy/classes/staging_model_database_records.py
@@ -96,7 +96,7 @@ class Legacy_StableDiffusion_ModelRecord(StagingLegacy_Generic_ModelRecord):
     homepage: str | None = None
     size_on_disk_bytes: int | None = None
     optimization: str | None = None
-    requirements: dict[str, int | str | list[int] | list[str] | bool] | None = None
+    requirements: dict[str, int | float | str | list[int] | list[float] | list[str] | bool] | None = None
 
 
 class Legacy_Generic_ModelReference(BaseModel):

diff --git a/horde_model_reference/model_reference_records.py b/horde_model_reference/model_reference_records.py
@@ -90,7 +90,7 @@ class StableDiffusion_ModelRecord(Generic_ModelRecord):
     style: MODEL_STYLE | str | None = None
     """The style of the model."""
 
-    requirements: dict[str, int | str | list[int] | list[str] | bool] | None = None
+    requirements: dict[str, int | float | str | list[int] | list[float] | list[str] | bool] | None = None
 
     size_on_disk_bytes: int | None = None
 

diff --git a/legacy_stable_diffusion.schema.json b/legacy_stable_diffusion.schema.json
@@ -242,6 +242,9 @@
                                     {
                                         "type": "integer"
                                     },
+                                    {
+                                        "type": "number"
+                                    },
                                     {
                                         "type": "string"
                                     },
@@ -251,6 +254,12 @@
                                         },
                                         "type": "array"
                                     },
+                                    {
+                                        "items": {
+                                            "type": "number"
+                                        },
+                                        "type": "array"
+                                    },
                                     {
                                         "items": {
                                             "type": "string"

diff --git a/stable_diffusion.schema.json b/stable_diffusion.schema.json
@@ -284,6 +284,9 @@
                                     {
                                         "type": "integer"
                                     },
+                                    {
+                                        "type": "number"
+                                    },
                                     {
                                         "type": "string"
                                     },
@@ -293,6 +296,12 @@
                                         },
                                         "type": "array"
                                     },
+                                    {
+                                        "items": {
+                                            "type": "number"
+                                        },
+                                        "type": "array"
+                                    },
                                     {
                                         "items": {
                                             "type": "string"