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

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

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: code-loader
-Version: 1.0.81a0
+Version: 1.0.82
 Summary: 
-Home-page: https://github.com/tensorleap/code-loader
 License: MIT
 Author: dorhar
 Author-email: doron.harnoy@tensorleap.ai
@@ -17,6 +16,7 @@ Requires-Dist: numpy (>=1.22.3,<2.0.0)
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
 Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
+Project-URL: Homepage, https://github.com/tensorleap/code-loader
 Project-URL: Repository, https://github.com/tensorleap/code-loader
 Description-Content-Type: text/markdown
 

{code_loader-1.0.81a0 → code_loader-1.0.82}/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.82}/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.82}/code_loader/leaploader.py

@@ -462,7 +462,7 @@ class LeapLoader(LeapLoaderBase):
 
         try:
             is_none = False
-            if value is None or (isinstance(value, (int, float, np.integer, np.floating)) and not np.isfinite(value)):
+            if value is None:
                 raise ValueError()
             converted_value = metadata_type_to_python_type[metadata_name_to_type[metadata_name]](value)
         except ValueError:

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

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