precommit

Author: kthyng

particle_tracking_manager/models/opendrift/config_opendrift.py

@@ -776,9 +776,17 @@ class HarmfulAlgalBloomModelConfig(HABParameters, OceanDriftModelConfig):
         default=HABSpeciesTypeEnum.PN,
         description="HarmfulAlgalBloom species category for this simulation. This option maps to individual properties which can instead be set manually if desired.",
         title="HAB Species Type",
-        json_schema_extra={"ptm_level": 1,
-                            "oneOf": [{"const": species.name, "title": species.value, "description": _species_descriptions[species.value]} for species in HABSpeciesTypeEnum],
-                        },
+        json_schema_extra={
+            "ptm_level": 1,
+            "oneOf": [
+                {
+                    "const": species.name,
+                    "title": species.value,
+                    "description": _species_descriptions[species.value],
+                }
+                for species in HABSpeciesTypeEnum
+            ],
+        },
     )
 
     # # new field: full parameter bundle for the HAB species

particle_tracking_manager/models/opendrift/enums/species_types.py

@@ -105,5 +105,15 @@ class HABParameters(BaseModel):
 }
 
 # this is for the schema
-_species_descriptions = {species.value: "Defaults: " + ", ".join([f"{key}={value}" for key, value in SPECIES_HAB_DEFAULTS[species].model_dump().items()]) for species in HABSpeciesTypeEnum if species != "custom"}
+_species_descriptions = {
+    species.value: "Defaults: "
+    + ", ".join(
+        [
+            f"{key}={value}"
+            for key, value in SPECIES_HAB_DEFAULTS[species].model_dump().items()
+        ]
+    )
+    for species in HABSpeciesTypeEnum
+    if species != "custom"
+}
 _species_descriptions["custom"] = "Custom species with user-defined parameters."