code-loader 1.0.84__tar.gz → 1.0.85__tar.gz

{code_loader-1.0.84 → code_loader-1.0.85}/PKG-INFO

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

{code_loader-1.0.84 → code_loader-1.0.85}/code_loader/contract/datasetclasses.py

@@ -7,7 +7,7 @@ import numpy.typing as npt
 from code_loader.contract.enums import DataStateType, DataStateEnum, LeapDataType, ConfusionMatrixValue, \
     MetricDirection, DatasetMetadataType
 from code_loader.contract.visualizer_classes import LeapImage, LeapText, LeapGraph, LeapHorizontalBar, \
-    LeapTextMask, LeapImageMask, LeapImageWithBBox, LeapImageWithHeatmap
+    LeapTextMask, LeapImageMask, LeapImageWithBBox, LeapImageWithHeatmap, LeapVideo
 
 custom_latent_space_attribute = "custom_latent_space"
 
@@ -94,6 +94,7 @@ class UnlabeledDataPreprocessHandler:
 
 VisualizerCallableInterface = Union[
     Callable[..., LeapImage],
+    Callable[..., LeapVideo],
     Callable[..., LeapText],
     Callable[..., LeapGraph],
     Callable[..., LeapHorizontalBar],

{code_loader-1.0.84 → code_loader-1.0.85}/code_loader/inner_leap_binder/leapbinder.py

@@ -20,7 +20,7 @@ from code_loader.utils import to_numpy_return_wrapper, get_shape
 from code_loader.visualizers.default_visualizers import DefaultVisualizer, \
     default_graph_visualizer, \
     default_image_visualizer, default_horizontal_bar_visualizer, default_word_visualizer, \
-    default_image_mask_visualizer, default_text_mask_visualizer, default_raw_data_visualizer
+    default_image_mask_visualizer, default_text_mask_visualizer, default_raw_data_visualizer, default_video_visualizer
 
 
 class LeapBinder:
@@ -48,6 +48,8 @@ class LeapBinder:
     def _extend_with_default_visualizers(self) -> None:
         self.set_visualizer(function=default_image_visualizer, name=DefaultVisualizer.Image.value,
                             visualizer_type=LeapDataType.Image)
+        self.set_visualizer(function=default_video_visualizer, name=DefaultVisualizer.Video.value,
+                            visualizer_type=LeapDataType.Video)
         self.set_visualizer(function=default_graph_visualizer, name=DefaultVisualizer.Graph.value,
                             visualizer_type=LeapDataType.Graph)
         self.set_visualizer(function=default_raw_data_visualizer, name=DefaultVisualizer.RawData.value,

{code_loader-1.0.84 → code_loader-1.0.85}/code_loader/visualizers/default_visualizers.py

@@ -4,12 +4,13 @@ import numpy as np
 import numpy.typing as npt
 
 from code_loader.contract.visualizer_classes import LeapImage, LeapGraph, LeapHorizontalBar, LeapText, \
-    LeapImageMask, LeapTextMask
+    LeapImageMask, LeapTextMask, LeapVideo
 from code_loader.utils import rescale_min_max
 
 
 class DefaultVisualizer(Enum):
     Image = 'Image'
+    Video = 'Video'
     Graph = 'Graph'
     HorizontalBar = 'HorizontalBar'
     Text = 'Text'
@@ -23,6 +24,10 @@ def default_image_visualizer(data: npt.NDArray[np.float32]) -> LeapImage:
     return LeapImage(rescaled_data)
 
 
+def default_video_visualizer(data: npt.NDArray[np.float32]) -> LeapVideo:
+    return LeapVideo(data)
+
+
 def default_graph_visualizer(data: npt.NDArray[np.float32]) -> LeapGraph:
     return LeapGraph(data)
 

{code_loader-1.0.84 → code_loader-1.0.85}/pyproject.toml

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