code-loader 1.0.52__py3-none-any.whl → 1.0.55__py3-none-any.whl

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] == leap_binder.batch_size_to_validate, \
-                        (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] == leap_binder.batch_size_to_validate, \
+                    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 = {
@@ -279,7 +279,7 @@ def tensorleap_gt_encoder(name: str):
     def decorating_function(user_function: SectionCallableInterface):
         for gt_handler in leap_binder.setup_container.ground_truths:
             if gt_handler.name == name:
-                raise Exception(f'Input with name {name} already exists. '
+                raise Exception(f'GT with name {name} already exists. '
                                 f'Please choose another')
 
         leap_binder.set_ground_truth(user_function, name)
@@ -319,7 +319,7 @@ def tensorleap_custom_loss(name: str):
     def decorating_function(user_function: CustomCallableInterface):
         for loss_handler in leap_binder.setup_container.custom_loss_handlers:
             if loss_handler.name == name:
-                raise Exception(f'Input with name {name} already exists. '
+                raise Exception(f'Custom loss with name {name} already exists. '
                                 f'Please choose another')
 
         leap_binder.add_custom_loss(user_function, name)
@@ -327,8 +327,8 @@ def tensorleap_custom_loss(name: str):
         def _validate_input_args(*args, **kwargs):
             try:
                 import tensorflow as tf
-            except ImportError:
-                raise Exception('the input arguments of the custom loss function should be tensorflow tensors')
+            except ImportError as e:
+                raise Exception('the input arguments of the custom loss function should be tensorflow tensors') from e
 
             for i, arg in enumerate(args):
                 assert isinstance(arg, tf.Tensor), (f'tensorleap_custom_loss validation failed: '
@@ -367,11 +367,11 @@ def tensorleap_custom_layer(name: str):
 
         try:
             import tensorflow as tf
-        except ImportError:
-            raise Exception('The custom layer should be inherited from tf.keras.layers.Layer')
+        except ImportError as e:
+            raise Exception('Custom layer should be inherit from tf.keras.layers.Layer') from e
 
         if not issubclass(custom_layer, tf.keras.layers.Layer):
-            raise Exception('The custom layer should be inherited from tf.keras.layers.Layer')
+            raise Exception('Custom layer should be inherit from tf.keras.layers.Layer')
 
         leap_binder.set_custom_layer(custom_layer, name)
 

{code_loader-1.0.52.dist-info → code_loader-1.0.55.dist-info}/METADATA

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

{code_loader-1.0.52.dist-info → code_loader-1.0.55.dist-info}/RECORD

@@ -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=iiyEls8CT0v2WseJ5nGgNaSVEfWm0GLgPsjjkKlLf8Y,19153
+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=VoqO9FN84yXyMjRjHjUTOt2GdTkJRMbHbXJ1cJkREkk,2230
-code_loader-1.0.52.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.52.dist-info/METADATA,sha256=aGRFw7VPjzKyFXyeth17cyCxFPCUN1wa14Ja7rLRPlM,888
-code_loader-1.0.52.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.52.dist-info/RECORD,,
+code_loader-1.0.55.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.55.dist-info/METADATA,sha256=I2IBhfNvtdxQgrU5SLkIpEjam8s4YcwG7f8coQ1TBhk,888
+code_loader-1.0.55.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.55.dist-info/RECORD,,