code-loader 1.0.91.dev11__tar.gz → 1.0.91.dev13__tar.gz

{code_loader-1.0.91.dev11 → code_loader-1.0.91.dev13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.91.dev11
+Version: 1.0.91.dev13
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-1.0.91.dev11 → code_loader-1.0.91.dev13}/code_loader/contract/datasetclasses.py

@@ -5,7 +5,7 @@ import numpy as np
 import numpy.typing as npt
 
 from code_loader.contract.enums import DataStateType, DataStateEnum, LeapDataType, ConfusionMatrixValue, \
-    MetricDirection, DatasetMetadataType
+    MetricDirection, DatasetMetadataType, InstanceFillingType
 from code_loader.contract.visualizer_classes import LeapImage, LeapText, LeapGraph, LeapHorizontalBar, \
     LeapTextMask, LeapImageMask, LeapImageWithBBox, LeapImageWithHeatmap, LeapVideo
 
@@ -72,6 +72,7 @@ class PreprocessResponse:
 class ElementInstance:
     name: str
     mask: npt.NDArray[np.float32]
+    # instance_filling_type: InstanceFillingType # TODO: implement InstanceFillingType
 
 SectionCallableInterface = Callable[[Union[int, str], PreprocessResponse], npt.NDArray[np.float32]]
 InstanceCallableInterface = Callable[[int, PreprocessResponse], List[ElementInstance]]

{code_loader-1.0.91.dev11 → code_loader-1.0.91.dev13}/code_loader/contract/enums.py

@@ -17,6 +17,10 @@ class MetricEnum(Enum):
 # should not be copied to engine contracts
 Metric = MetricEnum
 
+# class InstanceFillingType(Enum): # TODO: implement InstanceFillingType
+#     Zeros = 'Zeros'
+#     Blur = 'Blur'
+#     Learnable = 'Learnable'
 
 class LeapDataType(Enum):
     Image = 'Image'

{code_loader-1.0.91.dev11 → code_loader-1.0.91.dev13}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -279,9 +279,8 @@ def tensorleap_element_instance_preprocess(instance_mask_encoder: Callable[[int,
                 instance_to_sample_ids_mappings = {}
                 all_sample_ids = preprocess_response.sample_ids.copy()
                 for sample_id in preprocess_response.sample_ids:
-                    data_length = len(all_sample_ids)
                     instances_masks = instance_mask_encoder(sample_id, preprocess_response)
-                    instances_ids = list(range(data_length, data_length + len(instances_masks)))
+                    instances_ids = [f'{sample_id}_{instance_id}' for instance_id in range(len(instances_masks))]
                     sample_ids_to_instance_mappings[sample_id] = instances_ids
                     instance_to_sample_ids_mappings[sample_id] = sample_id
                     for instance_id in instances_ids:
@@ -294,6 +293,9 @@ def tensorleap_element_instance_preprocess(instance_mask_encoder: Callable[[int,
 
         leap_binder.set_preprocess(user_function_instance)
 
+        def metadata_sample_index(idx: str, preprocess: PreprocessResponse) -> bool:
+            return False
+        leap_binder.set_metadata(metadata_sample_index, "is_instance")
 
         def _validate_input_args(*args, **kwargs):
             assert len(args) == 0 and len(kwargs) == 0, \

{code_loader-1.0.91.dev11 → code_loader-1.0.91.dev13}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.91.dev11"
+version = "1.0.91.dev13"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"