code-loader 1.0.162__tar.gz → 1.0.164.dev1__tar.gz

{code_loader-1.0.162 → code_loader-1.0.164.dev1}/PKG-INFO

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

{code_loader-1.0.162 → code_loader-1.0.164.dev1}/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
 
@@ -693,14 +693,15 @@ def tensorleap_custom_metric(name: str,
                              f'Keys in the direction mapping should be part of result keys. Got key {direction_key}.')
 
                 if compute_insights is not None:
-                    assert isinstance(compute_insights, dict), \
+                    assert isinstance(compute_insights, (bool, dict)), \
                         (f'{user_function.__name__}() validation failed: '
-                         f'compute_insights should be dict if using the dict results. Got {type(compute_insights)}.')
+                         f'compute_insights should be bool or dict if using the dict results. Got {type(compute_insights)}.')
 
-                    for ci_key in compute_insights:
-                        assert ci_key in result, \
-                            (f'{user_function.__name__}() validation failed: '
-                             f'Keys in the compute_insights mapping should be part of result keys. Got key {ci_key}.')
+                    if isinstance(compute_insights, dict):
+                        for ci_key in compute_insights:
+                            assert ci_key in result, \
+                                (f'{user_function.__name__}() validation failed: '
+                                 f'Keys in the compute_insights mapping should be part of result keys. Got key {ci_key}.')
 
             else:
                 _validate_single_metric(result)
@@ -824,7 +825,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])
@@ -1687,8 +1689,12 @@ def tensorleap_status_table():
         mandatory_ready_mess = "\nAll mandatory parts have been successfully set. If no errors accured, you can now push the project to the Tensorleap system or continue to the next optional reccomeded interface,adding: "
         code_mapping_failure_mes = "Tensorleap_integration_test code flow failed, check raised exception."
 
-        name_width = max(len(row["name"]) for row in table)
-        status_width = max(len(row["Added to integration"]) for row in table)
+        # Sort table: CHECK first, then CROSS, then UNKNOWN
+        status_order = {CHECK: 0, CROSS: 1, UNKNOWN: 2}
+        sorted_table = sorted(table, key=lambda row: status_order.get(row["Added to integration"], 3))
+
+        name_width = max(len(row["name"]) for row in sorted_table)
+        status_width = max(len(row["Added to integration"]) for row in sorted_table)
 
         header = f"{'Decorator Name'.ljust(name_width)} | {'Added to integration'.ljust(status_width)}"
         sep = "-" * len(header)
@@ -1696,11 +1702,13 @@ def tensorleap_status_table():
         print("\n" + header)
         print(sep)
 
+        for row in sorted_table:
+            print(f"{row['name'].ljust(name_width)} | {row['Added to integration'].ljust(status_width)}")
+
+        # Determine next_step based on original table order (recommended integration order)
         ready = True
         next_step = None
-
         for row in table:
-            print(f"{row['name'].ljust(name_width)} | {row['Added to integration'].ljust(status_width)}")
             if not _crashed["value"] and ready:
                 if row["Added to integration"] != CHECK:
                     ready = False
@@ -1764,4 +1772,3 @@ if not _call_from_tl_platform:
 
 
 
-

{code_loader-1.0.162 → code_loader-1.0.164.dev1}/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.162 → code_loader-1.0.164.dev1}/pyproject.toml

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