code-loader 1.0.81a0__tar.gz → 1.0.83__tar.gz

{code_loader-1.0.81a0 → code_loader-1.0.83}/PKG-INFO

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

{code_loader-1.0.81a0 → code_loader-1.0.83}/code_loader/contract/enums.py

@@ -27,6 +27,7 @@ class LeapDataType(Enum):
     TextMask = 'TextMask'
     ImageWithBBox = 'ImageWithBBox'
     ImageWithHeatmap = 'ImageWithHeatmap'
+    Video = 'Video'
 
 
 class MetricDirection(Enum):

{code_loader-1.0.81a0 → code_loader-1.0.83}/code_loader/contract/visualizer_classes.py

@@ -53,6 +53,30 @@ class LeapImage:
         validate_type(type(self.compress), bool, 'compress flag must be a boolean')
 
 
+@dataclass
+class LeapVideo:
+    """
+    Visualizer representing a video for Tensorleap.
+
+    Attributes:
+    data (npt.NDArray[np.float32] | npt.NDArray[np.uint8]): The video data, shaped [T, H, W, C], where T is the number of frames.
+    type (LeapDataType): The data type, default is LeapDataType.Video.
+
+    Example:
+        video_data = np.random.rand(10, 100, 100, 3).astype(np.float32)
+        leap_video = LeapVideo(data=video_data)
+    """
+    data: Union[npt.NDArray[np.float32], npt.NDArray[np.uint8]]
+    type: LeapDataType = LeapDataType.Video
+
+    def __post_init__(self) -> None:
+        validate_type(self.type, LeapDataType.Video)
+        validate_type(type(self.data), np.ndarray)
+        validate_type(self.data.dtype, [np.uint8, np.float32])
+        validate_type(len(self.data.shape), 4, 'Video must be of shape 4')
+        validate_type(self.data.shape[3], [1, 3], 'Video channel must be either 3(rgb) or 1(gray)')
+
+
 @dataclass
 class LeapImageWithBBox:
     """
@@ -308,6 +332,7 @@ class LeapImageWithHeatmap:
 map_leap_data_type_to_visualizer_class = {
     LeapDataType.Image.value: LeapImage,
     LeapDataType.Graph.value: LeapGraph,
+    LeapDataType.Video.value: LeapVideo,
     LeapDataType.Text.value: LeapText,
     LeapDataType.HorizontalBar.value: LeapHorizontalBar,
     LeapDataType.ImageMask.value: LeapImageMask,

{code_loader-1.0.81a0 → code_loader-1.0.83}/pyproject.toml

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