code-loader 1.0.102__py3-none-any.whl → 1.0.102.dev1__py3-none-any.whl

code_loader/helpers/plot_functions.py

@@ -20,7 +20,21 @@ import math
 from code_loader.contract.visualizer_classes import LeapImage, LeapImageWithBBox, LeapGraph, LeapText, \
     LeapHorizontalBar, LeapImageMask, LeapTextMask, LeapImageWithHeatmap
 
+def run_only_on_non_mapping_mode():
+    """
+    Decorator to ensure that the function is only run when not in mapping mode.
+    """
+    def decorator(func):
+        def wrapper(*args, **kwargs):
+            if os.environ.get(mapping_runtime_mode_env_var_mame):
+                print(f"Skipping {func.__name__} in mapping mode.")
+                return
+            return func(*args, **kwargs)
+        return wrapper
+    return decorator
 
+
+@run_only_on_non_mapping_mode()
 def plot_image_with_b_box(leap_data: LeapImageWithBBox, title: str) -> None:
     """
     Plot an image with overlaid bounding boxes.
@@ -78,6 +92,7 @@ def plot_image_with_b_box(leap_data: LeapImageWithBBox, title: str) -> None:
     plt.show()
 
 
+@run_only_on_non_mapping_mode()
 def plot_image(leap_data: LeapImage, title: str) -> None:
     """
     Display the image contained in the LeapImage object.
@@ -108,6 +123,7 @@ def plot_image(leap_data: LeapImage, title: str) -> None:
     plt.show()
 
 
+@run_only_on_non_mapping_mode()
 def plot_graph(leap_data: LeapGraph, title: str) -> None:
     """
         Display the line chart contained in the LeapGraph object.
@@ -145,6 +161,7 @@ def plot_graph(leap_data: LeapGraph, title: str) -> None:
     plt.show()
 
 
+@run_only_on_non_mapping_mode()
 def plot_text_with_heatmap(leap_data: LeapText, title: str) -> None:
     """
     Display the text contained in the LeapText object with a heatmap overlay.
@@ -213,7 +230,7 @@ def plot_text_with_heatmap(leap_data: LeapText, title: str) -> None:
     plt.tight_layout()
     plt.show()
 
-
+@run_only_on_non_mapping_mode()
 def plot_hbar(leap_data: LeapHorizontalBar, title: str) -> None:
     """
         Display the horizontal bar chart contained in the LeapHorizontalBar object.
@@ -269,7 +286,7 @@ def plot_hbar(leap_data: LeapHorizontalBar, title: str) -> None:
 
     plt.show()
 
-
+@run_only_on_non_mapping_mode()
 def plot_image_mask(leap_data: LeapImageMask, title: str) -> None:
     """
         Plots an image with overlaid masks given a LeapImageMask visualizer object.
@@ -318,7 +335,7 @@ def plot_image_mask(leap_data: LeapImageMask, title: str) -> None:
     plt.axis('off')  # Hide the axis
     plt.show()
 
-
+@run_only_on_non_mapping_mode()
 def plot_text_mask(leap_data: LeapTextMask, title: str) -> None:
     """
         Plots text with overlaid masks given a LeapTextMask visualizer object.
@@ -370,7 +387,7 @@ def plot_text_mask(leap_data: LeapTextMask, title: str) -> None:
 
     plt.show()
 
-
+@run_only_on_non_mapping_mode()
 def plot_image_with_heatmap(leap_data: LeapImageWithHeatmap, title: str) -> None:
     """
         Display the image with overlaid heatmaps contained in the LeapImageWithHeatmap object.
@@ -413,6 +430,7 @@ def plot_image_with_heatmap(leap_data: LeapImageWithHeatmap, title: str) -> None
     plt.show()
 
 
+
 plot_switch = {
     LeapDataType.Image: plot_image,
     LeapDataType.Text: plot_text_with_heatmap,

code_loader/inner_leap_binder/leapbinder.py

@@ -49,6 +49,7 @@ class LeapBinder:
         self._extend_with_default_losses()
 
         self.mapping_connections: List[NodeConnection] = []
+        self.integration_test_func: Optional[Callable] = None
 
         self.batch_size_to_validate: Optional[int] = None
 

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -35,6 +35,34 @@ def _add_mapping_connections(connects_to, arg_names, node_mapping_type, name):
 
 
 
+def integration_test():
+    def decorating_function(integration_test_function: Callable):
+        leap_binder.integration_test_func = integration_test_function
+
+
+        def inner(*args, **kwargs):
+            ret = integration_test_function(*args, **kwargs)
+
+            try:
+                os.environ[mapping_runtime_mode_env_var_mame] = 'True'
+                integration_test_function(None, None)
+            except Exception as e:
+                print(f'Error during integration test: Make sure to disable any non tensorleap decorators '
+                      f'functions before pushing a new TL version')
+            finally:
+                if mapping_runtime_mode_env_var_mame in os.environ:
+                    del os.environ[mapping_runtime_mode_env_var_mame]
+
+
+        return inner
+
+    return decorating_function
+
+
+
+
+
+
 
 def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]] = None):
     for i, prediction_type in enumerate(prediction_types):

{code_loader-1.0.102.dist-info → code_loader-1.0.102.dev1.dist-info}/METADATA

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

{code_loader-1.0.102.dist-info → code_loader-1.0.102.dev1.dist-info}/RECORD

@@ -20,17 +20,17 @@ code_loader/experiment_api/types.py,sha256=MY8xFARHwdVA7p4dxyhD60ShmttgTvb4qdp1o
 code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZJEqBqc,3304
 code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
 code_loader/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/helpers/plot_functions.py,sha256=FHix9_iPiFvm8cYxT1ezQ58jqxo7_uY5_FhM8T4v79Q,13911
+code_loader/helpers/plot_functions.py,sha256=91wHSewc_y75hdCL-4ghrsGpg0u-IlK_7aoxbu6-jd0,14602
 code_loader/helpers/visualize.py,sha256=RusWyJax9qSkz2JItMb_IxOLskUvfGSWajTJHY4hCuE,565
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
-code_loader/inner_leap_binder/leapbinder.py,sha256=mi9wp98iywHedCe2GwrbiqE14zbGo1rR0huodG96ZXY,32508
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=j38nYWfc6yll1SMggV8gABEvSyQwEBVf5RdFnmQ1aD0,38523
+code_loader/inner_leap_binder/leapbinder.py,sha256=tiWrRBZCuQDeWOP0GGRUvzFHRFKG0uX0mWADdiSP-ho,32570
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=0-Itrq_rBzx4InpmcfkSmUX6lOmrzlqGKxtkAMD8H5I,39356
 code_loader/leaploader.py,sha256=vfN92-uoLeo8pojhwzPh4iu3gaoIQNqQklYwOy0kbtM,29225
 code_loader/leaploaderbase.py,sha256=lKdw2pd6H9hFsxVmc7jJMoZd_vlG5He1ooqT-cR_yq8,4496
 code_loader/utils.py,sha256=_j8b60pimoNAvWMRj7hEkkT6C76qES6cZoBFHpXHMxA,2698
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.102.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.102.dist-info/METADATA,sha256=eeUs000GUbSg8MZPX3Ly4wNc_8Y_1G53yCEqD-MNACk,850
-code_loader-1.0.102.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.102.dist-info/RECORD,,
+code_loader-1.0.102.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.102.dev1.dist-info/METADATA,sha256=QFOrHLfO7eTASzOF-Cyvu9Nf-pI5Rur-w27OlxES3lc,855
+code_loader-1.0.102.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.102.dev1.dist-info/RECORD,,