code-loader 1.0.52__tar.gz → 1.0.53__tar.gz

{code_loader-1.0.52 → code_loader-1.0.53}/PKG-INFO

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

{code_loader-1.0.52 → code_loader-1.0.53}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -93,7 +93,7 @@ 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, \
+                    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]}')
 
@@ -101,7 +101,7 @@ def tensorleap_custom_visualizer(name: str, visualizer_type: LeapDataType,
                 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] == 1, \
                         (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]}')
 
@@ -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 → code_loader-1.0.53}/code_loader/visualizers/default_visualizers.py

@@ -19,20 +19,32 @@ 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]
@@ -40,10 +52,17 @@ 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)]
 
@@ -57,6 +76,10 @@ 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.52 → code_loader-1.0.53}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.52"
+version = "1.0.53"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"