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

code_loader/helpers/plot_functions.py

@@ -1,12 +1,27 @@
-import matplotlib.pyplot as plt  # type: ignore
+
+import os
+
+from code_loader.inner_leap_binder.leapbinder import mapping_runtime_mode_env_var_mame
+
+if not os.environ.get(mapping_runtime_mode_env_var_mame):
+    try:
+        import matplotlib.pyplot as plt  # type: ignore
+    except ImportError:
+        raise ImportError(
+            "Matplotlib is not installed. Please install it using 'pip install matplotlib' to visualize Leap data."
+        )
+
+
 import numpy as np
-from code_loader.contract.enums import LeapDataType  # type: ignore
-from code_loader.contract.datasetclasses import LeapData  # type: ignore
+from code_loader.contract.enums import LeapDataType
 from textwrap import wrap
 import math
 
+from code_loader.contract.visualizer_classes import LeapImage, LeapImageWithBBox, LeapGraph, LeapText, \
+    LeapHorizontalBar, LeapImageMask, LeapTextMask, LeapImageWithHeatmap
+
 
-def plot_image_with_b_box(leap_data: LeapData, title: str) -> None:
+def plot_image_with_b_box(leap_data: LeapImageWithBBox, title: str) -> None:
     """
     Plot an image with overlaid bounding boxes.
 
@@ -63,7 +78,7 @@ def plot_image_with_b_box(leap_data: LeapData, title: str) -> None:
     plt.show()
 
 
-def plot_image(leap_data: LeapData, title: str) -> None:
+def plot_image(leap_data: LeapImage, title: str) -> None:
     """
     Display the image contained in the LeapImage object.
 
@@ -93,7 +108,7 @@ def plot_image(leap_data: LeapData, title: str) -> None:
     plt.show()
 
 
-def plot_graph(leap_data: LeapData, title: str) -> None:
+def plot_graph(leap_data: LeapGraph, title: str) -> None:
     """
         Display the line chart contained in the LeapGraph object.
 
@@ -130,7 +145,7 @@ def plot_graph(leap_data: LeapData, title: str) -> None:
     plt.show()
 
 
-def plot_text_with_heatmap(leap_data: LeapData, title: str) -> None:
+def plot_text_with_heatmap(leap_data: LeapText, title: str) -> None:
     """
     Display the text contained in the LeapText object with a heatmap overlay.
 
@@ -199,7 +214,7 @@ def plot_text_with_heatmap(leap_data: LeapData, title: str) -> None:
     plt.show()
 
 
-def plot_hbar(leap_data: LeapData, title: str) -> None:
+def plot_hbar(leap_data: LeapHorizontalBar, title: str) -> None:
     """
         Display the horizontal bar chart contained in the LeapHorizontalBar object.
 
@@ -255,7 +270,7 @@ def plot_hbar(leap_data: LeapData, title: str) -> None:
     plt.show()
 
 
-def plot_image_mask(leap_data: LeapData, title: str) -> None:
+def plot_image_mask(leap_data: LeapImageMask, title: str) -> None:
     """
         Plots an image with overlaid masks given a LeapImageMask visualizer object.
 
@@ -304,7 +319,7 @@ def plot_image_mask(leap_data: LeapData, title: str) -> None:
     plt.show()
 
 
-def plot_text_mask(leap_data: LeapData, title: str) -> None:
+def plot_text_mask(leap_data: LeapTextMask, title: str) -> None:
     """
         Plots text with overlaid masks given a LeapTextMask visualizer object.
 
@@ -356,7 +371,7 @@ def plot_text_mask(leap_data: LeapData, title: str) -> None:
     plt.show()
 
 
-def plot_image_with_heatmap(leap_data: LeapData, title: str) -> None:
+def plot_image_with_heatmap(leap_data: LeapImageWithHeatmap, title: str) -> None:
     """
         Display the image with overlaid heatmaps contained in the LeapImageWithHeatmap object.
 

code_loader/helpers/visualize.py

@@ -1,6 +1,6 @@
 import sys
 
-from code_loader.contract.datasetclasses import LeapData  # type: ignore
+from code_loader.contract.datasetclasses import LeapData
 
 from typing import Optional
 
@@ -15,5 +15,5 @@ def visualize(leap_data: LeapData, title: Optional[str] = None) -> None:
 
     if not title:
         title = f"Leap {leap_data.type.name} Visualization"
-    vis_function(leap_data, title)
+    vis_function(leap_data, title)  # type: ignore[operator]
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.101
+Version: 1.0.102
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT
@@ -13,7 +13,6 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Dist: matplotlib (>=3.3.4)
 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)

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

@@ -20,8 +20,8 @@ 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=QyCDxWrXmqAz9jbwOcDrRZ2xXAMmzwfr1CaY5X9wcTY,13426
-code_loader/helpers/visualize.py,sha256=AUd5OSJs4YhL_8LJ44MQECJhsBiYfGrLk269GWhnp58,555
+code_loader/helpers/plot_functions.py,sha256=FHix9_iPiFvm8cYxT1ezQ58jqxo7_uY5_FhM8T4v79Q,13911
+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
@@ -30,7 +30,7 @@ code_loader/leaploaderbase.py,sha256=lKdw2pd6H9hFsxVmc7jJMoZd_vlG5He1ooqT-cR_yq8
 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.101.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.101.dist-info/METADATA,sha256=-mfEuwf_Ltx_i9hrG3HhFwCibS1KYSvLcD3O6cwssAE,886
-code_loader-1.0.101.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.101.dist-info/RECORD,,
+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,,