fix: Conditionally import matplotlib

microdf/chart_utils.py

@@ -1,6 +1,3 @@
-import matplotlib as mpl
-
-
 def dollar_format(suffix=""):
     """Dollar formatter for matplotlib.
 
@@ -19,6 +16,12 @@ def currency_format(currency="USD", suffix=""):
     :returns: FuncFormatter.
 
     """
+    try:
+        import matplotlib as mpl
+    except ImportError:
+        raise ImportError(
+            "The function you've called requires extra dependencies. Please install microdf with the 'charts' extra by running 'pip install microdf[charts]'"
+        )
 
     prefix = {"USD": "$", "GBP": "£"}[currency]
 

microdf/charts.py

@@ -1,5 +1,3 @@
-import matplotlib as mpl
-import matplotlib.pyplot as plt
 import numpy as np
 
 import microdf as mdf
@@ -20,6 +18,8 @@ def quantile_pct_chg_plot(df1, df2, col1, col2, w1=None, w2=None, q=None):
     """
     try:
         import seaborn as sns
+        import matplotlib as mpl
+        import matplotlib.pyplot as plt
     except ImportError:
         raise ImportError(
             "The function you've called requires extra dependencies. Please install microdf with the 'charts' extra by running 'pip install microdf[charts]'"

microdf/style.py

@@ -1,7 +1,3 @@
-import matplotlib as mpl
-import matplotlib.font_manager as fm
-
-
 TITLE_COLOR = "#212121"
 AXIS_COLOR = "#757575"
 GRID_COLOR = "#eeeeee"  # Previously lighter #f5f5f5.
@@ -17,6 +13,8 @@ def set_plot_style(dpi: int = DPI):
     """
     try:
         import seaborn as sns
+        import matplotlib as mpl
+        import matplotlib.font_manager as fm
     except ImportError:
         raise ImportError(
             "The function you've called requires extra dependencies. Please install microdf with the 'charts' extra by running 'pip install microdf[charts]'"