From a3e23771cb43c403e02380d373af91d3db70df57 Mon Sep 17 00:00:00 2001
From: tazlin <tazlin.on.github@gmail.com>
Date: Mon, 30 Sep 2024 05:00:34 -0400
Subject: [PATCH] fix: set new type hint for other `requirements` fields

Also updates the automatically generated schema files
---
 .../legacy/classes/raw_legacy_model_database_records.py  | 2 +-
 .../legacy/classes/staging_model_database_records.py     | 2 +-
 horde_model_reference/model_reference_records.py         | 2 +-
 legacy_stable_diffusion.schema.json                      | 9 +++++++++
 stable_diffusion.schema.json                             | 9 +++++++++
 5 files changed, 21 insertions(+), 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
index 12c85eb..deecb3b 100644
--- 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
index 05171bb..95299a4 100644
--- 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
index 43ed171..ec31a4a 100644
--- 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
index 217f305..c9df766 100644
--- 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
index 79d120f..ad6edc9 100644
--- 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"