code-loader 1.0.91.dev6__py3-none-any.whl → 1.0.91.dev7__py3-none-any.whl

code_loader/contract/datasetclasses.py

@@ -248,3 +248,5 @@ class DatasetSample:
     metadata_is_none: Dict[str, bool]
     index: Union[int, str]
     state: DataStateEnum
+    instance_masks: Optional[Dict[str, List[npt.NDArray[np.float32]]]] = None
+

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -347,7 +347,7 @@ def tensorleap_unlabeled_preprocess():
     return decorating_function
 
 
-def tensorleap_instances_masks_encoder(name: str, model_input_index=None):
+def tensorleap_instances_masks_encoder(name: str):
     def decorating_function(user_function: InstanceCallableInterface):
         leap_binder.set_instance_masks(user_function, name)
 

code_loader/leaploader.py

@@ -150,7 +150,7 @@ class LeapLoader(LeapLoaderBase):
                                state=state)
         return sample
 
-    def get_sample_masks(self, state: DataStateEnum, sample_id: Union[int, str]) -> List[DatasetSample]:
+    def get_sample_with_masks(self, state: DataStateEnum, sample_id: Union[int, str]) -> DatasetSample:
         self.exec_script()
         preprocess_result = self._preprocess_result()
         if state == DataStateEnum.unlabeled and sample_id not in preprocess_result[state].sample_ids:
@@ -162,7 +162,8 @@ class LeapLoader(LeapLoaderBase):
                                metadata=metadata,
                                metadata_is_none=metadata_is_none,
                                index=sample_id,
-                               state=state)
+                               state=state,
+                               instance_masks=self._get_masks(state, sample_id))
         return sample
 
     def check_dataset(self) -> DatasetIntegParseResult:
@@ -452,6 +453,16 @@ class LeapLoader(LeapLoaderBase):
     def _get_inputs(self, state: DataStateEnum, sample_id: Union[int, str]) -> Dict[str, npt.NDArray[np.float32]]:
         return self._get_dataset_handlers(global_leap_binder.setup_container.inputs, state, sample_id)
 
+    def _get_masks(self, state: DataStateEnum, sample_id: Union[int, str]) -> Dict[str, List[npt.NDArray[np.float32]]]:
+        preprocess_result = self._preprocess_result()
+        preprocess_state = preprocess_result[state]
+        result_agg = {}
+        for handler in global_leap_binder.setup_container.instance_masks:
+            handler_result = handler.function(sample_id, preprocess_state)
+            handler_name = handler.name
+            result_agg[handler_name] = handler_result
+        return result_agg
+
     def _get_gt(self, state: DataStateEnum, sample_id: Union[int, str]) -> Dict[str, npt.NDArray[np.float32]]:
         return self._get_dataset_handlers(global_leap_binder.setup_container.ground_truths, state, sample_id)
 

{code_loader-1.0.91.dev6.dist-info → code_loader-1.0.91.dev7.dist-info}/METADATA

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

{code_loader-1.0.91.dev6.dist-info → code_loader-1.0.91.dev7.dist-info}/RECORD

@@ -1,7 +1,7 @@
 LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
 code_loader/__init__.py,sha256=6MMWr0ObOU7hkqQKgOqp4Zp3I28L7joGC9iCbQYtAJg,241
 code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/contract/datasetclasses.py,sha256=miWme96EUeRKzhVf2qPzpkr8y1JBTjH3ZcEx9lZmg20,8533
+code_loader/contract/datasetclasses.py,sha256=L8r4VN9x99gN8xTKl-JoO4r_h8Abx21XhtbZ0Xuw_f8,8612
 code_loader/contract/enums.py,sha256=GEFkvUMXnCNt-GOoz7NJ9ecQZ2PPDettJNOsxsiM0wk,1622
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
 code_loader/contract/mapping.py,sha256=e11h_sprwOyE32PcqgRq9JvyahQrPzwqgkhmbQLKLQY,1165
@@ -21,13 +21,13 @@ code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZ
 code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=ETTvjFHq6VX1zgKqyyVP3nJLGyUBrfO1uGF89B7sTYo,32807
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=_sYLXoi10wYUVsJ86bDUVYZgyTlcwE46HsR0qkaYEWg,28830
-code_loader/leaploader.py,sha256=BEmoW_tBXOJaXmbM_L8rdN9MsmjDDdqXqVSTfMtsW7w,28237
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=3T9sq9UmrRgfHuTJqJii0uyG6xd060cPKJ98TFFQcrc,28806
+code_loader/leaploader.py,sha256=GotVm-yi8xhSLEg4E8Qt7q0Yip24d73p-NRSTxzbFmU,28837
 code_loader/leaploaderbase.py,sha256=mUwQcOEiYFhevhUetyga4YdJXC1dAbntehW9i2jDl_Q,4558
 code_loader/utils.py,sha256=lISPFgQETSXc-P9jsIDq8YhXiaT58sHX0bXMJ4Aavqg,2722
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=669lBpLISLO6my5Qcgn1FLDDeZgHumPf252m4KHY4YM,2555
-code_loader-1.0.91.dev6.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.91.dev6.dist-info/METADATA,sha256=YvZW5SmQJ-xbeI-aAI1uYUvVyQneSjKDgSqeyYsOSGk,854
-code_loader-1.0.91.dev6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.91.dev6.dist-info/RECORD,,
+code_loader-1.0.91.dev7.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.91.dev7.dist-info/METADATA,sha256=XPKPLBCkCOpZ56A8jeaBNy25amVcJ0MKPLH2yWLdmZo,854
+code_loader-1.0.91.dev7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.91.dev7.dist-info/RECORD,,