code-loader 1.0.161.dev2__tar.gz → 1.0.163__tar.gz

{code_loader-1.0.161.dev2 → code_loader-1.0.163}/PKG-INFO

@@ -1,7 +1,8 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.161.dev2
+Version: 1.0.163
 Summary: 
+Home-page: https://github.com/tensorleap/code-loader
 License: MIT
 Author: dorhar
 Author-email: doron.harnoy@tensorleap.ai
@@ -19,7 +20,6 @@ Requires-Dist: numpy (>=2.3.2,<3.0.0) ; python_version >= "3.11" and python_vers
 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.161.dev2 → code_loader-1.0.163}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -24,7 +24,7 @@ from code_loader.contract.enums import MetricDirection, LeapDataType, DatasetMet
 from code_loader import leap_binder, LeapLoader
 from code_loader.contract.mapping import NodeMapping, NodeMappingType, NodeConnection
 from code_loader.contract.visualizer_classes import LeapImage, LeapImageMask, LeapTextMask, LeapText, LeapGraph, \
-    LeapHorizontalBar, LeapImageWithBBox, LeapImageWithHeatmap
+    LeapHorizontalBar, LeapImageWithBBox, LeapImageWithHeatmap, LeapVideo
 from code_loader.inner_leap_binder.leapbinder import mapping_runtime_mode_env_var_mame
 from code_loader.mixpanel_tracker import clear_integration_events, AnalyticsEvent, emit_integration_event_once
 
@@ -824,7 +824,8 @@ def tensorleap_custom_visualizer(name: str, visualizer_type: LeapDataType,
                 LeapDataType.Graph: LeapGraph,
                 LeapDataType.HorizontalBar: LeapHorizontalBar,
                 LeapDataType.ImageWithBBox: LeapImageWithBBox,
-                LeapDataType.ImageWithHeatmap: LeapImageWithHeatmap
+                LeapDataType.ImageWithHeatmap: LeapImageWithHeatmap,
+                LeapDataType.Video: LeapVideo
             }
             validate_output_structure(result, func_name=user_function.__name__,
                                       expected_type_name=result_type_map[visualizer_type])

{code_loader-1.0.161.dev2 → code_loader-1.0.163}/code_loader/plot_functions/plot_functions.py

@@ -20,7 +20,7 @@ from textwrap import wrap
 import math
 
 from code_loader.contract.visualizer_classes import LeapImage, LeapImageWithBBox, LeapGraph, LeapText, \
-    LeapHorizontalBar, LeapImageMask, LeapTextMask, LeapImageWithHeatmap
+    LeapHorizontalBar, LeapImageMask, LeapTextMask, LeapImageWithHeatmap, LeapVideo
 
 def run_only_on_non_mapping_mode():
     """
@@ -388,6 +388,18 @@ def plot_text_mask(leap_data: LeapTextMask, title: str) -> None:
 
     plt.show()
 
+@run_only_on_non_mapping_mode()
+def plot_video(leap_data: LeapVideo, title: str) -> None:
+    video_data = leap_data.data
+    fig, ax = plt.subplots()
+    fig.patch.set_facecolor('black')
+    ax.set_facecolor('black')
+    ax.set_title(title, color='white')
+    for frame in video_data:
+        ax.imshow(frame)
+        plt.pause(0.1)  # Adjust the pause duration as needed
+    plt.show()
+
 @run_only_on_non_mapping_mode()
 def plot_image_with_heatmap(leap_data: LeapImageWithHeatmap, title: str) -> None:
     """
@@ -441,4 +453,5 @@ plot_switch = {
     LeapDataType.TextMask: plot_text_mask,
     LeapDataType.ImageWithHeatmap: plot_image_with_heatmap,
     LeapDataType.ImageWithBBox: plot_image_with_b_box,
+    LeapDataType.Video: plot_video,
 }

{code_loader-1.0.161.dev2 → code_loader-1.0.163}/pyproject.toml

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