code-loader 1.0.73__py3-none-any.whl → 1.0.75__py3-none-any.whl

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -8,7 +8,7 @@ import numpy.typing as npt
 from code_loader.contract.datasetclasses import CustomCallableInterfaceMultiArgs, \
     CustomMultipleReturnCallableInterfaceMultiArgs, ConfusionMatrixCallableInterfaceMultiArgs, CustomCallableInterface, \
     VisualizerCallableInterface, MetadataSectionCallableInterface, PreprocessResponse, SectionCallableInterface, \
-    ConfusionMatrixElement
+    ConfusionMatrixElement, SamplePreprocessResponse
 from code_loader.contract.enums import MetricDirection, LeapDataType
 from code_loader import leap_binder
 from code_loader.contract.visualizer_classes import LeapImage, LeapImageMask, LeapTextMask, LeapText, LeapGraph, \
@@ -30,18 +30,18 @@ def tensorleap_custom_metric(name: str,
 
         def _validate_input_args(*args, **kwargs) -> None:
             for i, arg in enumerate(args):
-                assert isinstance(arg, np.ndarray), (f'tensorleap_custom_metric validation failed: '
+                assert isinstance(arg, (np.ndarray, SamplePreprocessResponse)), (f'tensorleap_custom_metric validation failed: '
                                                      f'Argument #{i} should be a numpy array. Got {type(arg)}.')
-                if leap_binder.batch_size_to_validate:
+                if leap_binder.batch_size_to_validate and isinstance(arg, np.ndarray):
                     assert arg.shape[0] == leap_binder.batch_size_to_validate, \
                         (f'tensorleap_custom_metric validation failed: Argument #{i} '
                          f'first dim should be as the batch size. Got {arg.shape[0]} '
                          f'instead of {leap_binder.batch_size_to_validate}')
 
             for _arg_name, arg in kwargs.items():
-                assert isinstance(arg, np.ndarray), (f'tensorleap_custom_metric validation failed: '
+                assert isinstance(arg, (np.ndarray, SamplePreprocessResponse)), (f'tensorleap_custom_metric validation failed: '
                                                      f'Argument {_arg_name} should be a numpy array. Got {type(arg)}.')
-                if leap_binder.batch_size_to_validate:
+                if leap_binder.batch_size_to_validate and isinstance(arg, np.ndarray):
                     assert arg.shape[0] == leap_binder.batch_size_to_validate, \
                         (f'tensorleap_custom_metric validation failed: Argument {_arg_name} '
                          f'first dim should be as the batch size. Got {arg.shape[0]} '
@@ -115,17 +115,17 @@ def tensorleap_custom_visualizer(name: str, visualizer_type: LeapDataType,
 
         def _validate_input_args(*args, **kwargs):
             for i, arg in enumerate(args):
-                assert isinstance(arg, np.ndarray), (f'tensorleap_custom_visualizer validation failed: '
+                assert isinstance(arg, (np.ndarray, SamplePreprocessResponse)), (f'tensorleap_custom_visualizer validation failed: '
                                                      f'Argument #{i} should be a numpy array. Got {type(arg)}.')
-                if leap_binder.batch_size_to_validate:
+                if leap_binder.batch_size_to_validate and isinstance(arg, np.ndarray):
                     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: '
+                assert isinstance(arg, (np.ndarray, SamplePreprocessResponse)), (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:
+                if leap_binder.batch_size_to_validate and isinstance(arg, np.ndarray):
                     assert arg.shape[0] != leap_binder.batch_size_to_validate, \
                         (f'tensorleap_custom_visualizer validation failed: Argument {_arg_name} '
                          f'should be without batch dimension. ')
@@ -353,10 +353,10 @@ def tensorleap_custom_loss(name: str):
 
         leap_binder.add_custom_loss(user_function, name)
 
-        valid_types = np.ndarray
+        valid_types = (np.ndarray, SamplePreprocessResponse)
         try:
             import tensorflow as tf
-            valid_types = (np.ndarray, tf.Tensor)
+            valid_types = (np.ndarray, SamplePreprocessResponse, tf.Tensor)
         except ImportError:
             pass
 

code_loader/leaploader.py

@@ -278,8 +278,8 @@ class LeapLoader(LeapLoaderBase):
 
         return vis_handler.function(**input_tensors_by_arg_name)
 
-    def run_heatmap_visualizer(self, visualizer_name: str, input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]],
-                               sample_ids: np.array, state: DataStateEnum,
+    def run_heatmap_visualizer(self, visualizer_name: str, sample_ids: np.array, state: DataStateEnum,
+                               input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]]
                                ) -> Optional[npt.NDArray[np.float32]]:
         heatmap_function = self._visualizer_handler_by_name()[visualizer_name].heatmap_function
         if heatmap_function is None:

{code_loader-1.0.73.dist-info → code_loader-1.0.75.dist-info}/METADATA

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

{code_loader-1.0.73.dist-info → code_loader-1.0.75.dist-info}/RECORD

@@ -20,13 +20,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=0l9zjlF27tZwg6SnpyqVoAAgf9QHQcKpR9lg7vho2Xw,27065
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=B-XSw4xYF39kMPnMTRNKMYFg09whnfl7VSbcx195VG8,21626
-code_loader/leaploader.py,sha256=he1c46jQNrCBEBq03gQDS0WVxiX7nlGAE9_9hLagxQc,24973
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=AI8Q9C_TiqpQqNmkCYZBeILFOaj1wn8LA64P8QbvNvs,21946
+code_loader/leaploader.py,sha256=Gggk5vzomqVqlYo14sQTlc7RIzhONYR-11hTLDKhwsA,24972
 code_loader/leaploaderbase.py,sha256=VH0vddRmkqLtcDlYPCO7hfz1_VbKo43lUdHDAbd4iJc,4198
 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=Ffx5VHVOe5ujBOsjBSxN_aIEVwFSQ6gbhTMG5aUS-po,2305
-code_loader-1.0.73.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.73.dist-info/METADATA,sha256=d7-FSiS0YvfmoWlJZyqlf644v1SuNGf-_MfunY5KYcA,849
-code_loader-1.0.73.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.73.dist-info/RECORD,,
+code_loader-1.0.75.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.75.dist-info/METADATA,sha256=gs1VXUV1uIfBxojPyzvlbGGHkGE647DzwASp1HJiNHU,849
+code_loader-1.0.75.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.75.dist-info/RECORD,,