code-loader 1.0.101__tar.gz → 1.0.102__tar.gz

{code_loader-1.0.101 → code_loader-1.0.102}/PKG-INFO

@@ -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 → code_loader-1.0.102}/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-1.0.101 → code_loader-1.0.102}/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 → code_loader-1.0.102}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.101"
+version = "1.0.102"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"
@@ -17,7 +17,6 @@ numpy = "^1.22.3"
 psutil = "^5.9.5"
 requests = "^2.32.3"
 pyyaml = "^6.0.2"
-matplotlib = ">= 3.3.4"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.1.1"