endoreg-db 0.8.4.4__py3-none-any.whl → 0.8.8.0__py3-none-any.whl

endoreg_db/data/requirement/colon_polyp_intervention.yaml

@@ -0,0 +1,49 @@
+- model: endoreg_db.requirement
+  fields:
+    name: "colonoscopy_default_polyp_resection"
+    requirement_types:
+      - "patient_finding"
+    operators:
+      - "models_match_any"
+    finding_interventions:
+      - "colon_lesion_biopsy"
+      - "colon_lesion_polypectomy_cold_snare"
+      - "colon_lesion_polypectomy_hot_snare"
+      - "colon_lesion_emr"
+      - "colon_lesion_esd"
+      - "colon_lesion_eftr"
+
+- model: endoreg_db.requirement
+  fields:
+    name: "colonoscopy_default_polyp_no_resection"
+    requirement_types:
+      - "patient_finding"
+    operators:
+      - "models_match_none"
+    finding_interventions:
+      - "colon_lesion_biopsy"
+      - "colon_lesion_polypectomy_cold_snare"
+      - "colon_lesion_polypectomy_hot_snare"
+      - "colon_lesion_emr"
+      - "colon_lesion_esd"
+      - "colon_lesion_eftr"
+# - model: endoreg_db.requirement
+#   fields:
+#     name: "colonoscopy_finding_colon_polyp_no_resection_reason"
+#     requirement_types:
+#       - "patient_finding"
+#     operators:
+#       - "models_match_any"
+#     finding_classifications:
+#       - "colon_lesion_no_resection_reason"
+
+# - model: endoreg_db.requirement
+#   fields:
+#     name: "colonoscopy_polyp_clip_application"
+#     requirement_types:
+#       - "patient_finding"
+#     operators:
+#       - "models_match_any"
+#     finding_interventions:
+#       - "colon_lesion_over_the_scope_clip"
+#       - "colon_lesion_clip"

endoreg_db/data/requirement/coloreg_colon_polyp.yaml

@@ -0,0 +1,49 @@
+- model: endoreg_db.requirement
+  fields:
+    name: "colonoscopy_default_size"
+    requirement_types:
+      - "patient_finding"
+    operators:
+      - "models_match_any"
+    finding_classifications:
+      - "colon_lesion_size_mm"
+      - "colon_lesion_size_categorical"
+
+- model: endoreg_db.requirement
+  fields:
+    name: "colonoscopy_default_location"
+    requirement_types:
+      - "patient_finding"
+    operators:
+      - "models_match_any"
+    finding_classifications:
+      - "colonoscopy_location_default"
+
+- model: endoreg_db.requirement
+  fields:
+    name: "colonoscopy_default_polyp_morphology"
+    operators:
+      - "models_match_any"
+    requirement_types:
+      - "patient_finding"
+    finding_classifications:
+      - "colon_lesion_paris"
+      - "colon_lesion_lst"
+      - "colon_lesion_planarity_default"
+
+- model: endoreg_db.requirement
+  fields:
+    name: "colonoscopy_default_polyp_histology"
+    requirement_types:
+      - "patient_finding"
+    operators:
+      - "models_match_all"
+    finding_classifications:
+      - "colonoscopy_polyp_histology"
+      - "colonoscopy_polyp_histology_dysplasia"
+      - "colonoscopy_histology_dignity"
+      - "colonoscopy_default_histology_in_sano"
+
+- models: endoreg_db.requirement
+  fields:
+    name: "colonoscopy_finding_colon_polyp_histo_expected"

endoreg_db/data/requirement_set/01_endoscopy_generic.yaml

@@ -2,7 +2,7 @@
   fields:
     name: "patient_gender_generic"
     name_de: "Patientengeschlecht (männlich/weiblich)"
-    name_en: "Patient gender generic" 
+    name_en: "Patient gender generic"
     description: "Patient gender (generic)"
     requirement_set_type: "any"
     requirements:
@@ -11,17 +11,35 @@
 
 - models: endoreg_db.requirement_set
   fields:
-    name: "basic_lab_values_normal"
-    name_de: "Basis-Labortests (Unauffällig)"
-    name_en: "Basic Lab Tests (Normal)"
-    description: "Basic lab tests (normal)"
+    name: "patient_names_generic"
+    name_de: "Patientennamen (Vorname, Nachname)"
+    name_en: "Patient names generic"
+    description: "Patient names (generic)"
     requirement_set_type: "all"
     requirements:
-      - "lab_value_hb_normal"
-      - "lab_value_leukocytes_normal"
-      - "lab_value_platelets_normal"
-      - "lab_value_creatinine_normal"
-      - "lab_value_inr_normal"
+      - patient_first_name #TODO
+      - patient_last_name #TODO
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "patient_dob_generic"
+    name_de: "Patienten-Geburtsdatum"
+    name_en: "Patient date of birth generic"
+    description: "Patient date of birth (generic)"
+    requirement_set_type: "all"
+    requirements:
+      - patient_date_of_birth_present #TODO
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "colonoscopy_basic_patient_history"
+    name_de: "Koloskopie - Basis Patientenanamnese"
+    name_en: "Colonoscopy - Basic Patient History"
+    description: "Colonoscopy - Basic Patient History"
+    requirement_set_type: "all"
+    requirements:
+      - "patient_history_last_colonoscopy_known" #TODO
+      - "patient_history_previous_bowel_surgery_known" #TODO
 
 - models: endoreg_db.requirement_set
   fields:
@@ -34,7 +52,6 @@
       - "patient_age_gte_18"
       - "patient_age_lte_100"
 
-# - models: endoreg_db.requirement_set
 - models: endoreg_db.requirement_set
   fields:
     name: "generic_patient"
@@ -45,4 +62,6 @@
     links_to_sets:
       - "patient_age_generic"
       - "patient_gender_generic"
-      - "basic_lab_values_normal"
+      - "basic_lab_values_normal"
+
+-

endoreg_db/data/requirement_set/01_laboratory.yaml

@@ -0,0 +1,13 @@
+- models: endoreg_db.requirement_set
+  fields:
+    name: "basic_lab_values_normal"
+    name_de: "Basis-Labortests (Unauffällig)"
+    name_en: "Basic Lab Tests (Normal)"
+    description: "Basic lab tests (normal)"
+    requirement_set_type: "all"
+    requirements:
+      - "lab_value_hb_normal"
+      - "lab_value_leukocytes_normal"
+      - "lab_value_platelets_normal"
+      - "lab_value_creatinine_normal"
+      - "lab_value_inr_normal"

endoreg_db/data/requirement_set/02_endoscopy_bleeding_risk.yaml

@@ -0,0 +1,46 @@
+- models: endoreg_db.requirement_set
+  fields:
+    name: "endoscopy_bleeding_risk_high"
+    name_de: "Endoskopie - Blutungsrisiko: hoch"
+    name_en: "Endoscopy - Bleeding Risk: High"
+    description: "Endoskopie - Blutungsrisiko: hoch"
+    # linked_sets: [] # if this rule depends on others
+    requirement_set_type: "any"
+    requirements:
+      - "endoscopy_intervention_bleeding_risk_high"
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "endoscopy_bleeding_risk_low"
+    name_de: "Endoskopie - Blutungsrisiko: niedrig"
+    name_en: "Endoscopy - Bleeding Risk: Low"
+    description: "Endoskopie - Blutungsrisiko: niedrig"
+    # linked_sets: [] # if this rule depends on others
+    requirement_set_type: "any"
+    requirements:
+      - "endoscopy_intervention_bleeding_risk_low"
+    links_to_sets: []
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "patient_medication_indication_anticoagulation"
+    name_de: "Patientenmedikation - Antikoagulation"
+    name_en: "Patient Medication - Anticoagulation"
+    description: "Patient medication - anticoagulation"
+    requirement_set_type: "any"
+    requirements:
+      - "indication_non_valvular_af_apixaban_is_defined_in_requirement"
+      - "indication_after_hip_apixaban_is_defined_in_requirement"
+      - "indication_after_knee_apixaban_is_defined_in_requirement"
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "high_bleed_risk_endoscopy_w_patient_w_anticoagulation"
+    name_de: "Endoskopie - hohes Blutungsrisiko mit generischen Patientenanforderungen"
+    name_en: "High Bleed Risk Endoscopy with Generic Patient Requirements"
+    description: "High bleed risk endoscopy with generic patient requirements"
+    requirement_set_type: "all"
+    links_to_sets:
+      - "endoscopy_bleeding_risk_high"
+      - "generic_patient"
+      - "patient_medication_indication_anticoagulation"

endoreg_db/data/requirement_set/90_coloreg.yaml

@@ -0,0 +1,178 @@
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_patient_data"
+    name_de: "Coloreg Patientendaten"
+    name_en: "Coloreg Patient Data"
+    description: "Coloreg Patient Data"
+    requirement_set_type: "all"
+    requirements: []
+    links_to_sets:
+      - "patient_gender_generic"
+      - "patient_names_generic"
+      - "patient_dob_generic"
+      - "colonoscopy_basic_patient_history" # includes previous bowel_surgery_known and last_colonoscopy_known
+    tags: [] # not tagged as report_mask_requirement_set, only used in top level requirement set
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_examination_data"
+    name_de: "Coloreg Untersuchungsdaten"
+    name_en: "Coloreg Examination Data"
+    description: "Coloreg Examination Data"
+    requirement_set_type: "all"
+    requirements:
+      - "examination_has_indication" #TODO
+      - "examination_has_sedation_information" #TODO
+      - "examination_bowel_prep_recorded" #TODO
+    links_to_sets:
+      - "colonoscopy_deepest_insertion_reached_recorded" #TODO # if not cecum or terminal ileum
+    tags: []
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_location_base"
+    name_de: "Coloreg Befund Kolon Polyp Lage"
+    name_en: "Coloreg Finding Colon Polyp Location"
+    description: "Coloreg Finding Colon Polyp Exact Location"
+    requirement_set_type: "any"
+    requirements:
+      - "colon_polyp_has_location" #TODO
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_rectosigmoid_location"
+    name_de: "Coloreg Befund Kolon Polyp Rektosigmoid Lage"
+    name_en: "Coloreg Finding Colon Polyp Rectosigmoid Location"
+    description: "Coloreg Finding Colon Polyp Rectosigmoid Location"
+    requirement_set_type: "all"
+    depends_on_sets:
+      - "coloreg_finding_colon_polyp_location_base"
+    requirements:
+      - "coloreg_finding_colon_polyp_rectosigmoid_has_location_cm" #TODO Rektosigmoid muss cm haben
+      - "coloreg_finding_colon_polyp_non_rectosigmoid_has_location_cm" #TODO nicht rektosigmoid kann cm haben
+      - "coloreg_finding_colon_polyp_non_rectosigmoid_has_location_colon_segment" #TODO nicht rektosigmoid kann Segment haben
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_size"
+    name_de: "Coloreg Befund Kolon Polyp Größe"
+    name_en: "Coloreg Finding Colon Polyp Size"
+    description: "Coloreg Finding Colon Polyp Size"
+    requirement_set_type: "any"
+    requirements:
+      - "coloreg_finding_colon_polyp_size_measured_in_mm" #TODO
+      - "coloreg_finding_colon_polyp_size_categorical" #TODO
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_polyp_small_morphology"
+    name_de: "Coloreg Polyp Klein Morphologie"
+    name_en: "Coloreg Polyp Small Morphology"
+    description: "Coloreg Polyp Small Morphology"
+    requirement_set_type: "all"
+    requirements:
+      - "colon_polyp_size_smaller_10_mm" # any size_cat is <6 or 6 - 9 mm or size_mm is < 10 mm
+      - "colonoscopy_default_polyp_morphology_wli" # any (paris / description)
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_polyp_medium_morphology"
+    name_de: "Coloreg Polyp Mittel Morphologie"
+    name_en: "Coloreg Polyp Medium Morphology"
+    description: "Coloreg Polyp Medium Morphology"
+    requirement_set_type: "all"
+    requirements:
+      - "colon_polyp_size_smaller_10_19_mm" # size cat is 10-19 mm or size_mm between 10 and 19 mm
+      - "colonoscopy_default_polyp_morphology_wli"
+      - "colonoscopy_nbi_polyp_morphology" # (nice)
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_polyp_large_morphology"
+    name_de: "Coloreg Polyp Groß Morphologie"
+    name_en: "Coloreg Polyp Large Morphology"
+    description: "Coloreg Polyp Large Morphology"
+    requirement_set_type: "all"
+    requirements:
+      - "colon_polyp_size_20_mm_or_larger" # size cat is 20+ mm or size_mm >=20 mm
+      - "colonoscopy_nbi_polyp_morphology" # (nice)
+      - "colonoscopy_large_polyp_morphology" # ( LST )
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_morphology"
+    name_de: "Coloreg Befund Kolon Polyp Morphologie"
+    name_en: "Coloreg Finding Colon Polyp Morphology"
+    description: "Coloreg Finding Colon Polyp Morphology"
+    requirement_set_type: "any"
+    depends_on_sets:
+      - "coloreg_finding_colon_polyp_size"
+    links_to_sets:
+      - "coloreg_polyp_small_morphology"
+      - "coloreg_polyp_medium_morphology"
+      - "coloreg_polyp_large_morphology"
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_histology_expected_and_available"
+    name_de: "Coloreg Befund Kolon Polyp Histologie"
+    name_en: "Coloreg Finding Colon Polyp Histology"
+    description: "Coloreg Finding Colon Polyp Histology"
+    requirement_set_type: "all"
+    requirements:
+      - "colonoscopy_default_polyp_resection"
+      - "colonoscopy_default_polyp_histology"
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_histology_not_expected"
+    name_de: "Coloreg Befund Kolon Polyp Histologie"
+    name_en: "Coloreg Finding Colon Polyp Histology"
+    description: "Coloreg Finding Colon Polyp Histology"
+    requirement_set_type: "all"
+    requirements:
+      - "colonoscopy_default_polyp_no_resection"
+      # - "colonoscopy_finding_colon_polyp_no_resection_reason"
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_histology"
+    name_de: "Coloreg Befund Kolon Polyp Histologie"
+    name_en: "Coloreg Finding Colon Polyp Histology"
+    description: "Coloreg Finding Colon Polyp Histology"
+    requirement_set_type: "any"
+    links_to_sets:
+      - "coloreg_finding_colon_polyp_histology_expected_and_available"
+      - "coloreg_finding_colon_polyp_histology_not_expected"
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_base_description"
+    name_de: "Coloreg Befund Kolon Polyp"
+    name_en: "Coloreg Finding Colon Polyp"
+    description: "Coloreg Finding Colon Polyp"
+    requirement_set_type: "all"
+    requirements: []
+    links_to_sets:
+      - "coloreg_finding_colon_polyp_size"
+      - "coloreg_finding_colon_polyp_location_base"
+      - "coloreg_finding_colon_polyp_rectosigmoid_location"
+      - "coloreg_finding_colon_polyp_morphology"
+      - "coloreg_finding_colon_polyp_histology"
+    tags: []
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "colonoscopy_coloreg"
+    name_de: "Koloskopie Coloreg"
+    name_en: "Colonoscopy Coloreg"
+    description: "Colonoscopy Coloreg"
+    requirement_set_type: "all"
+    links_to_sets:
+      - "coloreg_patient_data"
+      - "coloreg_examination_data"
+      - "coloreg_finding_colon_polyp_base_description"
+    tags:
+      - "report_mask_requirement_set"
+    reqset_exam_links:
+      - "colonoscopy_req_sets"

endoreg_db/data/requirement_set/_old_

@@ -0,0 +1,109 @@
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_patient_data"
+    name_de: "Coloreg Patientendaten"
+    name_en: "Coloreg Patient Data"
+    description: "Coloreg Patient Data"
+    requirement_set_type: "all"
+    requirements: []
+    links_to_sets:
+      - "patient_gender_generic"
+      - "patient_names_generic"
+      - "patient_dob_generic"
+      - "colonoscopy_basic_patient_history" # includes previous bowel_surgery_known and last_colonoscopy_known
+    tags: [] # not tagged as report_mask_requirement_set, only used in top level requirement set
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_examination_data"
+    name_de: "Coloreg Untersuchungsdaten"
+    name_en: "Coloreg Examination Data"
+    description: "Coloreg Examination Data"
+    requirement_set_type: "all"
+    requirements:
+      - "examination_has_indication" #TODO
+      - "examination_has_sedation_information" #TODO
+      - "examination_bowel_prep_recorded" #TODO
+    links_to_sets:
+      - "colonoscopy_deepest_insertion_reached_recorded" #TODO # if not cecum or terminal ileum
+    tags: []
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_location"
+    name_de: "Coloreg Befund Kolon Polyp Lage"
+    name_en: "Coloreg Finding Colon Polyp Location"
+    description: "Coloreg Finding Colon Polyp Exact Location"
+    requirement_set_type: "any"
+    requirements:
+      - "coloreg_finding_colon_polyp_rectosigmoid_has_location_distance_from_anal_verge" #TODO
+      - "coloreg_finding_colon_polyp_non_rectosigmoid_has_location_distance_from_anal_verge" #TODO
+      - "coloreg_finding_colon_polyp_non_rectosigmoid_has_location_colon_segment" #TODO
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_size"
+    name_de: "Coloreg Befund Kolon Polyp Größe"
+    name_en: "Coloreg Finding Colon Polyp Size"
+    description: "Coloreg Finding Colon Polyp Size"
+    requirement_set_type: "any"
+    requirements:
+      - "coloreg_finding_colon_polyp_size_measured_in_mm" #TODO
+      - "coloreg_finding_colon_polyp_size_categorical" #TODO
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_small_base"
+    name_de: "Coloreg Befund Kolon Polyp Klein"
+    name_en: "Coloreg Finding Colon Polyp Small"
+    description: "Coloreg Finding Colon Polyp Small"
+    requirement_set_type: "all"
+    requirements:
+      - "coloreg_finding_colon_polyp_size_smaller_10_mm" #TODO any; size_cat or size_mm
+      - "coloreg_finding_colon_polyp_location" # either location_categorical or location_cm (any)
+      - "colonoscopy_default_polyp_morphology_wli" # any (paris / description)
+      - "colonoscopy_default_polyp_histology" # all
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_medium_base"
+    name_de: "Coloreg Befund Kolon Polyp Mittel"
+    name_en: "Coloreg Finding Colon Polyp Medium"
+    description: "Coloreg Finding Colon Polyp Medium"
+    requirement_set_type: "all"
+    requirements:
+      - "coloreg_finding_colon_polyp_size_between_10_and_19_mm" #TODO size >=10mm
+      - "coloreg_finding_colon_polyp_location"
+      - "colonoscopy_default_polyp_morphology_wli"
+      - "colonoscopy_advanced_polyp_morphology"
+      - "colonoscopy_default_polyp_histology"
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "coloreg_finding_colon_polyp_base_description"
+    name_de: "Coloreg Befund Kolon Polyp"
+    name_en: "Coloreg Finding Colon Polyp"
+    description: "Coloreg Finding Colon Polyp"
+    requirement_set_type: "any"
+    requirements: []
+    links_to_sets:
+      - "coloreg_finding_colon_polyp_small_base" #TODO size <10mm, requires size, location, paris / description, histology
+      - "coloreg_finding_colon_polyp_medium_base" #TODO size >=10mm, requires size, location, paris / description, advanced morphology, histology
+      - "coloreg_finding_colon_polyp_large_base" #TODO size >=20mm, requires size, location, LST / description, advanced morphology, histology
+    tags: []
+
+- models: endoreg_db.requirement_set
+  fields:
+    name: "colonoscopy_coloreg"
+    name_de: "Koloskopie Coloreg"
+    name_en: "Colonoscopy Coloreg"
+    description: "Colonoscopy Coloreg"
+    requirement_set_type: "all"
+    links_to_sets:
+      - "coloreg_patient_data"
+      - "coloreg_examination_data"
+      - "coloreg_finding_colon_polyp_base_description"
+    tags:
+      - "report_mask_requirement_set"
+    reqset_exam_links:
+      - "colonoscopy_req_sets"

endoreg_db/data/requirement_set_type/data.yaml

@@ -18,3 +18,24 @@
     name_de: "keine"
     name_en: "none"
     description: "No requirement is true"
+
+- model: endoreg_db.requirement_set_type
+  fields:
+    name: "exactly_1"
+    name_de: "genau eins"
+    name_en: "exactly one"
+    description: "Exactly one requirement is true"
+
+- model: endoreg_db.requirement_set_type
+  fields:
+    name: "at_least_1"
+    name_de: "mindestens eins"
+    name_en: "at least one"
+    description: "At least one requirement is true"
+
+- model: endoreg_db.requirement_set_type
+  fields:
+    name: "at_most_1"
+    name_de: "höchstens eins"
+    name_en: "at most one"
+    description: "At most one requirement is true"

endoreg_db/data/setup_config.yaml

@@ -9,15 +9,15 @@ default_models:
 huggingface_fallback:
   enabled: true
   repo_id: "wg-lux/colo_segmentation_RegNetX800MF_base"
-  filename: "colo_segmentation_RegNetX800MF_base.ckpt"
+  filename: "colo_segmentation_RegNetX800MF_base.safetensors"
   labelset_name: "multilabel_classification_colonoscopy_default"
 
 # Local weights search patterns
 weights_search_patterns:
   # Primary model weight filename patterns (supports wildcards)
-  - "colo_segmentation_RegNetX800MF_*.ckpt"
-  - "image_multilabel_classification_colonoscopy_default_*.ckpt"
-  - "*_colonoscopy_*.ckpt"
+  - "colo_segmentation_RegNetX800MF_*.safetensors"
+  - "image_multilabel_classification_colonoscopy_default_*.safetensors"
+  - "*_colonoscopy_*.safetensors"
 
 # Search directories for local weights (relative to project root)
 weights_search_dirs:

endoreg_db/data/tag/requirement_set_tags.yaml

@@ -9,3 +9,24 @@
 - model: endoreg_db.tag
   fields:
     name: "colonoscopy_austria"
+
+# User role tags for requirement set filtering
+- model: endoreg_db.tag
+  fields:
+    name: "Gastroenterologist"
+
+- model: endoreg_db.tag
+  fields:
+    name: "Student"
+
+- model: endoreg_db.tag
+  fields:
+    name: "Professor"
+
+- model: endoreg_db.tag
+  fields:
+    name: "Terminology Expert"
+
+- model: endoreg_db.tag
+  fields:
+    name: "Default Prototype"

endoreg_db/exceptions.py

@@ -1,3 +1,6 @@
+import logging
+
+logger = logging.getLogger(__name__)
 
 
 class InsufficientStorageError(Exception):
@@ -7,7 +10,7 @@ class InsufficientStorageError(Exception):
         super().__init__(message)
         self.required_space = required_space
         self.available_space = available_space
-
+        
 
 class TranscodingError(Exception):
     """Raised when video transcoding fails."""
@@ -16,4 +19,4 @@ class TranscodingError(Exception):
 
 class VideoProcessingError(Exception):
     """Base class for video processing errors."""
-    pass
+    pass

endoreg_db/helpers/data_loader.py

@@ -12,7 +12,7 @@ def load_default_ai_model():
     This function constructs the path to the default AI model checkpoint and invokes the
     Django management command `create_multilabel_model_meta` to register the model.
     """
-    model_path = f"{ASSET_DIR.as_posix()}/colo_segmentation_RegNetX800MF_6.ckpt"
+    model_path = f"{ASSET_DIR.as_posix()}/colo_segmentation_RegNetX800MF_6.safetensors"
     # Pass arguments individually to call_command
     call_command(
         "create_multilabel_model_meta",