code-loader 1.0.184.dev5__tar.gz → 1.0.185__tar.gz

{code_loader-1.0.184.dev5 → code_loader-1.0.185}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: code-loader
-Version: 1.0.184.dev5
+Version: 1.0.185
 Summary: 
 License: MIT
 Author: dorhar

{code_loader-1.0.184.dev5 → code_loader-1.0.185}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -1189,13 +1189,19 @@ def _warn_image_mask(leap_data, viz_name: str, leap_type_name: str) -> None:
                      f"H x W {tuple(image.shape[:2])}. The mask overlay will fail or be misaligned."),
             link_to_docs=_VIZ_DOCS,
         )
+    # plot_image_mask colors label index i where mask == (i + 1), so valid class ids are
+    # 1..len(labels); id 0 is background (uncolored) and ids above len(labels) are never
+    # colored. Compare the actual id range, not the count (which mis-fires when labels is a
+    # global class list and only a subset of classes appears in this sample).
     nonzero = np.unique(mask)
     nonzero = nonzero[nonzero != 0]
-    if len(nonzero) != len(labels):
+    if nonzero.size > 0 and int(nonzero.max()) > len(labels):
         store_general_warning(
             key=("viz_mask_labels", viz_name, leap_type_name),
-            message=(f"{_viz_prefix(viz_name, leap_type_name)} mask has {len(nonzero)} distinct non-zero value(s) "
-                     f"but {len(labels)} label(s) were provided. Some regions may be uncolored or some labels unused."),
+            message=(f"{_viz_prefix(viz_name, leap_type_name)} mask contains class id(s) up to "
+                     f"{int(nonzero.max())} but only {len(labels)} label(s) were provided. plot_image_mask "
+                     f"colors ids 1..{len(labels)}, so regions with higher ids will be left uncolored. "
+                     f"Provide a label for each class id."),
             link_to_docs=_VIZ_DOCS,
         )
 

{code_loader-1.0.184.dev5 → code_loader-1.0.185}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "code-loader"
 
-version = "1.0.184.dev5"
+version = "1.0.185"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"