update imports to make matplotlib & seaborn optional

ymahlich · ymahlich · commit 5f3414a6b075 · 2025-08-04T10:45:52.000-07:00
diff --git a/coderdata/__init__.py b/coderdata/__init__.py
@@ -15,4 +15,12 @@
 from .utils.utils import version
 from .utils.utils import list_datasets
 
-from .utils.stats import summarize_response_metric
+try:
+    import matplotlib
+    import seaborn as sns
+except ModuleNotFoundError:
+    pass
+else:
+    from .utils.stats import summarize_response_metric
+    from .utils.stats import plot_response_metric
+    from .utils.stats import plot_2d_respones_metric
diff --git a/coderdata/utils/__init__.py b/coderdata/utils/__init__.py
@@ -1,4 +1,17 @@
-from .stats import summarize_response_metric
-from .stats import prepare_2d_hist_data
 from .utils import version
 from .utils import list_datasets
+
+try:
+    import matplotlib
+    import seaborn as sns
+except ModuleNotFoundError:
+    import warnings
+    warnings.warn(
+        "package was not availble. To use coderdata.utils.stats functions "
+        "please make sure 'matplotlib' & 'seaborn' are available in the "
+        "environment."
+        )
+else:
+    from .stats import summarize_response_metric
+    from .stats import plot_response_metric
+    from .stats import plot_2d_respones_metric
