[Bug] Decreased sensitivity of homogeneity checker

valab-certh · Apr 15, 2024 · d82a275 · d82a275
1 parent c829ed1
commit d82a275
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/main.py b/python/main.py
@@ -683,7 +683,7 @@ def check_nifti_integrity() -> bool:
             intensity_value_set = set(
                 np.unique(nib.load(nifti_fp).get_fdata().astype(np.uint8)),  # type: ignore[attr-defined]
             ).union(intensity_value_set)
-        return intensity_value_set == class_idcs
+        return intensity_value_set.issubset(class_idcs)
 
     homogeneity_state = check_class_names_integrity()
     if not homogeneity_state: