move ml_tools.extra.plotting to visualization.utils

sentinel-hub · Sep 6, 2023 · 7b893b1 · 7b893b1
1 parent cd119c1
commit 7b893b1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 11 deletions.
diff --git a/eolearn/ml_tools/extra/__init__.py b/eolearn/ml_tools/extra/__init__.py
diff --git a/eolearn/ml_tools/extra/plotting.py → eolearn/visualization/utils.py b/eolearn/ml_tools/extra/plotting.py → eolearn/visualization/utils.py
@@ -1,5 +1,5 @@
 """
-The module provides some utility functions for ML specific plotting
+The module provides some utility functions for plotting
 
 Copyright (c) 2017- Sinergise and contributors
 For the full list of contributors, see the CREDITS file in the root directory of this source tree.
@@ -11,13 +11,9 @@
 
 import itertools
 
+import matplotlib.pyplot as plt
 import numpy as np
-
-try:
-    import matplotlib.pyplot as plt
-    from matplotlib.colors import Colormap
-except ImportError as exception:
-    raise ImportError("This module requires an installation of matplotlib package") from exception
+from matplotlib.colors import Colormap
 
 
 def plot_confusion_matrix(

diff --git a/pyproject.toml b/pyproject.toml
@@ -70,7 +70,6 @@ io = ["affine", "fiona>=1.8.18", "rasterio>=1.2.7, <1.3.8"]
 mask = ["opencv-python-headless"]
 mask_extra = ["s2cloudless"]
 mltools = ["shapely"]
-mltoolsplotting = ["matplotlib"]
 visualization = ["graphviz>=0.10.1", "jinja2", "matplotlib", "pygments"]
 docs = [
     "eo-learn[ALL]",