fix: undo A003 style "fix"

Haidra-Org · Jul 20, 2024 · b0eaa98 · b0eaa98
1 parent 0acb6bc
commit b0eaa98
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/horde_model_reference/legacy/add_to_legacy_sd.py b/horde_model_reference/legacy/add_to_legacy_sd.py
@@ -250,7 +250,7 @@ def add_models_from_disk(self, file_out: pathlib.Path) -> None:
                 new_model_record = RawLegacy_StableDiffusion_ModelRecord(
                     name=model_friendly_name,
                     baseline=baseline_type,
-                    type_="ckpt",
+                    type="ckpt",
                     inpainting=inpainting,
                     description=model_description,
                     tags=tags_chosen if len(tags_chosen) > 0 else None,

diff --git a/horde_model_reference/legacy/classes/legacy_converters.py b/horde_model_reference/legacy/classes/legacy_converters.py
@@ -384,7 +384,7 @@ def parse_record(
                 self.all_styles.get(model_record_in_progress.style, 0) + 1
             )
 
-        if model_record_in_progress.type_ != "ckpt":
+        if model_record_in_progress.type != "ckpt":
             error = f"{model_record_key} is not a ckpt!"
             self.add_validation_error_to_log(model_record_key=model_record_key, error=error)
 

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
@@ -44,7 +44,7 @@ class RawLegacy_StableDiffusion_ModelRecord(BaseModel):
 
     name: str
     baseline: str
-    type_: str
+    type: str
     inpainting: bool
     description: str | None = None
     tags: list[str] | 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
@@ -53,7 +53,7 @@ class StagingLegacy_Generic_ModelRecord(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
     name: str
-    type_: str
+    type: str
     description: str | None = None
     version: str | None = None
     style: str | None = None