code-loader 1.0.112.dev3__py3-none-any.whl → 1.0.112.dev5__py3-none-any.whl

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -578,11 +578,18 @@ def tensorleap_instances_masks_encoder(name: str):
                 (f'tensorleap_instances_masks_encoder validation failed: '
                  f'Argument sample_id should be as the same type as defined in the preprocess response '
                  f'{preprocess_response.sample_id_type}. Got {type(sample_id)}.')
+            assert isinstance(instance_id, str), \
+                (f'tensorleap_instances_masks_encoder validation failed: '
+                 f'Argument instance_id should be str. Got {type(sample_id)}.')
+            assert type(instance_id) == preprocess_response.sample_id_type, \
+                (f'tensorleap_instances_masks_encoder validation failed: '
+                 f'Argument instance_id should be as the same type as defined in the preprocess response '
+                 f'{preprocess_response.sample_id_type}. Got {type(instance_id)}.')
 
         def _validate_result(result):
-            assert isinstance(result, ElementInstance), \
+            assert isinstance(result, ElementInstance) or (result is None), \
                 (f'tensorleap_instances_masks_encoder validation failed: '
-                 f'Unsupported return type. Should be a ElementInstance. Got {type(result)}.')
+                 f'Unsupported return type. Should be a ElementInstance or None. Got {type(result)}.')
 
         def inner_without_validate(sample_id, preprocess_response, instance_id):
             global _called_from_inside_tl_decorator

code_loader/utils.py

@@ -19,8 +19,10 @@ def to_numpy_return_wrapper(encoder_function: SectionCallableInterface) -> Secti
     return numpy_encoder_function
 
 def to_numpy_return_masks_wrapper(encoder_function: InstanceCallableInterface) -> InstanceCallableInterface:
-    def numpy_encoder_function(idx: Union[int, str], samples: PreprocessResponse, element_idx: Union[int, str]) -> ElementInstance:
+    def numpy_encoder_function(idx: Union[int, str], samples: PreprocessResponse, element_idx: Union[int, str]) -> Union[ElementInstance, None]:
         result = encoder_function(idx, samples, element_idx)
+        if result is None:
+            return None
         result.mask = np.array(result.mask)
         return result
     return numpy_encoder_function

{code_loader-1.0.112.dev3.dist-info → code_loader-1.0.112.dev5.dist-info}/METADATA

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

{code_loader-1.0.112.dev3.dist-info → code_loader-1.0.112.dev5.dist-info}/RECORD

@@ -21,16 +21,16 @@ 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=0iHVHxC2NjfH7F0vQFVGy1e0llgKEyUHUHh3DdtqL70,32602
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=EqM2x6mJxWysuWnZzfLMHOgnGtnfmRuHt-hKaXDXTcc,42882
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=0-K898pyrt7B9ziHV5npL6vcyydVoC-lSRnnczn6UMw,43460
 code_loader/leaploader.py,sha256=nIrUtK7n8is1MiaHS5T6io3P64brc1we5IxI4EPRqSs,29387
 code_loader/leaploaderbase.py,sha256=gvNP_MPjG3Ey8TB-efQI2s_gcGRXKlP1-qFDwOT80Mc,4510
 code_loader/plot_functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/plot_functions/plot_functions.py,sha256=xg6Gi4myTN9crq6JtyrhYI38HLXjPVJcbnI7CIy8f7w,14625
 code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6LznoQIvi4vpo,657
-code_loader/utils.py,sha256=jyKC7GSW92DEd0l1jxnNnYqTK8QzULjzlUV61Id17tc,2660
+code_loader/utils.py,sha256=i1KOchLkieHfaVz6YskSIfKA45HcqAmGAK1F2Kcg38c,2724
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.112.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.112.dev3.dist-info/METADATA,sha256=CCXtYeTb4uasde0qp1Bqi22jiJ9TPt_vO0RjECxhCJk,906
-code_loader-1.0.112.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.112.dev3.dist-info/RECORD,,
+code_loader-1.0.112.dev5.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.112.dev5.dist-info/METADATA,sha256=TLGeV1X6meziLDhwMSTZqXxzb_QsiotZQAlEefCxf90,906
+code_loader-1.0.112.dev5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.112.dev5.dist-info/RECORD,,