code-loader 1.0.53__py3-none-any.whl → 1.0.56__py3-none-any.whl

code_loader/contract/datasetclasses.py

@@ -49,6 +49,9 @@ class PreprocessResponse:
         else:
             raise Exception("length is deprecated.")
 
+    def __hash__(self) -> int:
+        return id(self)
+
     def __len__(self) -> int:
         assert self.sample_ids is not None
         return len(self.sample_ids)

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -93,17 +93,17 @@ def tensorleap_custom_visualizer(name: str, visualizer_type: LeapDataType,
                 assert isinstance(arg, np.ndarray), (f'tensorleap_custom_visualizer validation failed: '
                                                      f'Argument #{i} should be a numpy array. Got {type(arg)}.')
                 if leap_binder.batch_size_to_validate:
-                    assert arg.shape[0] == 1, \
-                        (f'tensorleap_custom_visualizer validation failed: Argument #{i} '
-                         f'first dim should be 1. The visualizers will always run with batch size 1. Got {arg.shape[0]}')
+                    assert arg.shape[0] != leap_binder.batch_size_to_validate, \
+                        (f'tensorleap_custom_visualizer validation failed: '
+                         f'Argument #{i} should be without batch dimension. ')
 
             for _arg_name, arg in kwargs.items():
                 assert isinstance(arg, np.ndarray), (f'tensorleap_custom_visualizer validation failed: '
                                                      f'Argument {_arg_name} should be a numpy array. Got {type(arg)}.')
                 if leap_binder.batch_size_to_validate:
-                    assert arg.shape[0] == 1, \
+                    assert arg.shape[0] != leap_binder.batch_size_to_validate, \
                         (f'tensorleap_custom_visualizer validation failed: Argument {_arg_name} '
-                         f'first dim should be 1. The visualizers will always run with batch size 1. Got {arg.shape[0]}')
+                         f'should be without batch dimension. ')
 
         def _validate_result(result):
             result_type_map = {

code_loader/visualizers/default_visualizers.py

@@ -19,32 +19,20 @@ class DefaultVisualizer(Enum):
     
 
 def default_image_visualizer(data: npt.NDArray[np.float32]) -> LeapImage:
-    # remove batch dim
-    data = data[0]
-
     rescaled_data = rescale_min_max(data)
     return LeapImage(rescaled_data)
 
 
 def default_graph_visualizer(data: npt.NDArray[np.float32]) -> LeapGraph:
-    # remove batch dim
-    data = data[0]
-
     return LeapGraph(data)
 
 
 def default_horizontal_bar_visualizer(data: npt.NDArray[np.float32]) -> LeapHorizontalBar:
-    # remove batch dim
-    data = data[0]
-
     labels = [str(index) for index in range(data.shape[-1])]
     return LeapHorizontalBar(data, labels)
 
 
 def default_word_visualizer(data: npt.NDArray[np.float32]) -> LeapText:
-    # remove batch dim
-    data = data[0]
-
     if hasattr(data, 'tolist'):
         data = data.tolist()
     words = [str(index[0]) if type(index) is list else str(index) for index in data]
@@ -52,17 +40,10 @@ def default_word_visualizer(data: npt.NDArray[np.float32]) -> LeapText:
 
 
 def default_raw_data_visualizer(data: npt.NDArray[np.float32]) -> LeapText:
-    # remove batch dim
-    data = data[0]
-
     return LeapText([str(data)])
 
 
 def default_image_mask_visualizer(mask: npt.NDArray[np.float32], image: npt.NDArray[np.float32]) -> LeapImageMask:
-    # remove batch dim
-    mask = mask[0]
-    image = image[0]
-
     n_different_labels = mask.shape[-1]
     labels = [str(i) for i in range(n_different_labels)]
 
@@ -76,10 +57,6 @@ def default_image_mask_visualizer(mask: npt.NDArray[np.float32], image: npt.NDAr
 
 
 def default_text_mask_visualizer(mask: npt.NDArray[np.float32], text_data: npt.NDArray[np.float32]) -> LeapTextMask:
-    # remove batch dim
-    mask = mask[0]
-    text_data = text_data[0]
-
     words = default_word_visualizer(text_data).data
     n_different_labels = mask.shape[-1]
     labels = [str(i) for i in range(n_different_labels)]

{code_loader-1.0.53.dist-info → code_loader-1.0.56.dist-info}/METADATA

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

{code_loader-1.0.53.dist-info → code_loader-1.0.56.dist-info}/RECORD

@@ -2,7 +2,7 @@ LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
 code_loader/__init__.py,sha256=6MMWr0ObOU7hkqQKgOqp4Zp3I28L7joGC9iCbQYtAJg,241
 code_loader/code_inegration_processes_manager.py,sha256=XslWOPeNQk4RAFJ_f3tP5Oe3EgcIR7BE7Y8r9Ty73-o,3261
 code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/contract/datasetclasses.py,sha256=lOIY-h9t4k9NxNsC9GrJhltmhpqRju3AuLA3WVQcCMs,6614
+code_loader/contract/datasetclasses.py,sha256=lFS7_weizsjzx4_tYwYGrrRUj1sgIl010h9FON4brb8,6670
 code_loader/contract/enums.py,sha256=6Lo7p5CUog68Fd31bCozIuOgIp_IhSiPqWWph2k3OGU,1602
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
 code_loader/contract/responsedataclasses.py,sha256=w7xVOv2S8Hyb5lqyomMGiKAWXDTSOG-FX1YW39bXD3A,3969
@@ -19,12 +19,12 @@ 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=35hyesDdmjOD9wdrTLyayb-vm9aDfmEbMA0c4EQR1LA,25090
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=F1qjyA0A1dFIUGT-YFnlR9YNHBS_bcLzjP5jxzmZXQw,19102
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=bBzIPx9DvPGzVTZcjUqCRkTT5GhH6cL5PsjmTtawJhE,19054
 code_loader/leaploader.py,sha256=POUgD6x1GH_iF_eDGz-VLX4DsIl2kddufKVDdrA_K-U,19491
 code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/visualizers/default_visualizers.py,sha256=pSNeucXepyyYFspHQbkV1nDGfpk4JwnMEGSOsWbLrj4,2581
-code_loader-1.0.53.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.53.dist-info/METADATA,sha256=jO5pY2aiuga9_e_CjGOIIafEPxXbMAeeEfeDnWyTMlU,888
-code_loader-1.0.53.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.53.dist-info/RECORD,,
+code_loader/visualizers/default_visualizers.py,sha256=VoqO9FN84yXyMjRjHjUTOt2GdTkJRMbHbXJ1cJkREkk,2230
+code_loader-1.0.56.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.56.dist-info/METADATA,sha256=ZgIrcewHLd99934-nSSMl3it13sgMLHLXhOj9jCPmSY,888
+code_loader-1.0.56.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.56.dist-info/RECORD,,