endoreg-db 0.8.8.9__py3-none-any.whl → 0.8.9.10__py3-none-any.whl

endoreg_db/utils/setup_config.py

@@ -68,7 +68,12 @@ class SetupConfig:
                 "image_multilabel_classification_colonoscopy_default_*.safetensors",
                 "*_colonoscopy_*.safetensors",
             ],
-            "weights_search_dirs": ["tests/assets", "assets", "data/storage/model_weights", "${STORAGE_DIR}/model_weights"],
+            "weights_search_dirs": [
+                "tests/assets",
+                "assets",
+                "data/storage/model_weights",
+                "${STORAGE_DIR}/model_weights",
+            ],
             "auto_generation_defaults": {
                 "activation": "sigmoid",
                 "mean": "0.485,0.456,0.406",
@@ -83,11 +88,16 @@ class SetupConfig:
 
     def get_primary_model_name(self) -> str:
         """Get the primary classification model name."""
-        return self._config.get("default_models", {}).get("primary_classification_model", "image_multilabel_classification_colonoscopy_default")
+        return self._config.get("default_models", {}).get(
+            "primary_classification_model",
+            "image_multilabel_classification_colonoscopy_default",
+        )
 
     def get_primary_labelset_name(self) -> str:
         """Get the primary labelset name."""
-        return self._config.get("default_models", {}).get("primary_labelset", "multilabel_classification_colonoscopy_default")
+        return self._config.get("default_models", {}).get(
+            "primary_labelset", "multilabel_classification_colonoscopy_default"
+        )
 
     def get_huggingface_config(self) -> Dict[str, Any]:
         """Get HuggingFace fallback configuration."""
@@ -97,7 +107,10 @@ class SetupConfig:
         """Get weight file search patterns."""
         return self._config.get(
             "weights_search_patterns",
-            ["colo_segmentation_RegNetX800MF_*.safetensors", "*_colonoscopy_*.safetensors"],
+            [
+                "colo_segmentation_RegNetX800MF_*.safetensors",
+                "*_colonoscopy_*.safetensors",
+            ],
         )
 
     def get_weights_search_dirs(self) -> List[Path]:
@@ -168,7 +181,10 @@ class SetupConfig:
 
                 if isinstance(data, list):
                     for item in data:
-                        if item.get("fields", {}).get("name") == model_name or item.get("fields", {}).get("model") == model_name:
+                        if (
+                            item.get("fields", {}).get("name") == model_name
+                            or item.get("fields", {}).get("model") == model_name
+                        ):
                             return item.get("setup_config", {})
         except Exception as e:
             logger.warning(f"Error loading model-specific config for {model_name}: {e}")

endoreg_db/utils/storage.py

@@ -60,7 +60,9 @@ def ensure_local_file(
                 shutil.copyfileobj(source, tmp_file, length=chunk_size)
         except Exception as exc:
             temp_path.unlink(missing_ok=True)
-            raise IOError(f"Could not download {field_file.name} from storage to a local file") from exc
+            raise IOError(
+                f"Could not download {field_file.name} from storage to a local file"
+            ) from exc
 
     try:
         yield temp_path

endoreg_db/utils/translation.py

@@ -5,23 +5,27 @@ def build_multilingual_response(obj, include_choices=False, classification_id=No
     If classification_id is given, adds 'classificationId' to each choice.
     """
     data = {
-        'id': obj.id,
-        'name': getattr(obj, 'name', ''),
-        'name_de': getattr(obj, 'name_de', ''),
-        'name_en': getattr(obj, 'name_en', ''),
-        'description': getattr(obj, 'description', ''),
-        'description_de': getattr(obj, 'description_de', ''),
-        'description_en': getattr(obj, 'description_en', ''),
+        "id": obj.id,
+        "name": getattr(obj, "name", ""),
+        "name_de": getattr(obj, "name_de", ""),
+        "name_en": getattr(obj, "name_en", ""),
+        "description": getattr(obj, "description", ""),
+        "description_de": getattr(obj, "description_de", ""),
+        "description_en": getattr(obj, "description_en", ""),
     }
-    if hasattr(obj, 'required'):
-        data['required'] = getattr(obj, 'required', True)
+    if hasattr(obj, "required"):
+        data["required"] = getattr(obj, "required", True)
     if include_choices:
-        data['choices'] = [
-            build_multilingual_response(choice, include_choices=False, classification_id=classification_id or obj.id)
+        data["choices"] = [
+            build_multilingual_response(
+                choice,
+                include_choices=False,
+                classification_id=classification_id or obj.id,
+            )
             for choice in obj.get_choices()
         ]
-        for choice_dict in data['choices']:
-            choice_dict['classificationId'] = classification_id or obj.id
+        for choice_dict in data["choices"]:
+            choice_dict["classificationId"] = classification_id or obj.id
     if classification_id is not None and not include_choices:
-        data['classificationId'] = classification_id
-    return data
+        data["classificationId"] = classification_id
+    return data

endoreg_db/utils/uuid.py

@@ -1,4 +1,5 @@
 import uuid
 
+
 def get_uuid():
     return uuid.uuid4()

endoreg_db/utils/validate_endo_roi.py

@@ -12,13 +12,21 @@ def validate_endo_roi(endo_roi_dict: Union[Dict[str, int], None]):
 
     for key, value in endo_roi_dict.items():
         if key == "x":
-            assert value >= 0, f"Endoscope ROI x value must be greater than or equal to 0. Got {value}"
+            assert value >= 0, (
+                f"Endoscope ROI x value must be greater than or equal to 0. Got {value}"
+            )
         elif key == "y":
-            assert value >= 0, f"Endoscope ROI y value must be greater than or equal to 0. Got {value}"
+            assert value >= 0, (
+                f"Endoscope ROI y value must be greater than or equal to 0. Got {value}"
+            )
         elif key == "width":
-            assert value > 0, f"Endoscope ROI width value must be greater than 0. Got {value}"
+            assert value > 0, (
+                f"Endoscope ROI width value must be greater than 0. Got {value}"
+            )
         elif key == "height":
-            assert value > 0, f"Endoscope ROI height value must be greater than 0. Got {value}"
+            assert value > 0, (
+                f"Endoscope ROI height value must be greater than 0. Got {value}"
+            )
         else:
             raise ValueError(f"Endoscope ROI key {key} not recognized")
 

endoreg_db/utils/validate_subcategory_dict.py

@@ -1,57 +1,58 @@
 from endoreg_db.models import Unit
 
-def validate_subcategory_dict(self, subcategory_dict:dict=None):
+
+def validate_subcategory_dict(self, subcategory_dict: dict = None):
     if subcategory_dict is None:
         return False
-    
+
     if not isinstance(subcategory_dict, dict):
         return False
 
     # check if key choices exists and is a list of strings
     if "choices" not in subcategory_dict:
         return False
-    
+
     if not isinstance(subcategory_dict["choices"], list):
         return False
-    
+
     for choice in subcategory_dict["choices"]:
         if not isinstance(choice, str):
             return False
-        
+
     # check if key default exists and is a string
     if "default" not in subcategory_dict:
         return False
-    
+
     if not isinstance(subcategory_dict["default"], str):
         return False
-    
+
     # check if default is in choices
     if subcategory_dict["default"] not in subcategory_dict["choices"]:
         return False
-    
+
     if not subcategory_dict["required"]:
         return False
-    
+
     return True
 
 
-def validate_numerical_descriptor(self, numerical_descriptor_dict:dict=None):
+def validate_numerical_descriptor(self, numerical_descriptor_dict: dict = None):
     # Check if numerical_descriptor_dict is None
     error_message = ""
     if numerical_descriptor_dict is None:
         error_message = "numerical_descriptor_dict is None"
         return False, error_message
-    
+
     # Check if numerical_descriptor_dict is a dictionary
     if not isinstance(numerical_descriptor_dict, dict):
         error_message = "numerical_descriptor_dict is not a dictionary"
         return False, error_message
-    
+
     # if key unit exists and is a string and Unit object with that name exists
     if "unit" not in numerical_descriptor_dict:
         error_message = "unit key does not exist in numerical_descriptor_dict"
         return False, error_message
-    
+
     elif not isinstance(numerical_descriptor_dict["unit"], str):
         error_message = "unit key is not a string"
         return False, error_message
@@ -59,33 +60,34 @@ def validate_numerical_descriptor(self, numerical_descriptor_dict:dict=None):
     elif not Unit.objects.filter(name=numerical_descriptor_dict["unit"]).exists():
         error_message = "Unit object with that name does not exist"
         return False, error_message
-    
-    
-    if not "required" in numerical_descriptor_dict:
+
+    if "required" not in numerical_descriptor_dict:
         error_message = "required key does not exist in numerical_descriptor_dict"
         return False, error_message
-    
+
     elif not isinstance(numerical_descriptor_dict["required"], bool):
         error_message = "required key is not a boolean"
         return False, error_message
-    
+
     # check if min, max, mean, std exist and are either None or float
     for key in ["min", "max", "mean", "std", "default"]:
         if key not in numerical_descriptor_dict:
             error_message = f"{key} key does not exist in numerical_descriptor_dict"
             return False, error_message
-        
-        if numerical_descriptor_dict[key] is not None and not isinstance(numerical_descriptor_dict[key], float):
+
+        if numerical_descriptor_dict[key] is not None and not isinstance(
+            numerical_descriptor_dict[key], float
+        ):
             error_message = f"{key} key is not a float"
             return False, error_message
-        
+
     # check if distribution exists and is either "normal" or "uniform"
     if "distribution" not in numerical_descriptor_dict:
         error_message = "distribution key does not exist in numerical_descriptor_dict"
         return False, error_message
-    
+
     if numerical_descriptor_dict["distribution"] not in ["normal", "uniform"]:
         error_message = "distribution key is not either 'normal' or 'uniform'"
         return False, error_message
-    
-    return True, None
+
+    return True, None