matplotlib optional

Author: alisaab

l0bnb/viz.py

@@ -1,7 +1,8 @@
-import matplotlib.pyplot as plt
-
-
 def graph_plot(x, y, x_axis, y_axis, title, loglog=True):
+    try:
+        import matplotlib.pyplot as plt
+    except ModuleNotFoundError:
+        raise Exception('matplotlib is not installed')
     plt.figure()
     func = plt.plot if not loglog else plt.loglog
     func(x, y, color='r', linestyle='-')
@@ -12,4 +13,8 @@ def graph_plot(x, y, x_axis, y_axis, title, loglog=True):
 
 
 def show_plots():
+    try:
+        import matplotlib.pyplot as plt
+    except ModuleNotFoundError:
+        raise Exception('matplotlib is not installed')
     plt.show()

setup.py

@@ -14,10 +14,9 @@ def readme():
     author="Hussein Hazimeh, Rahul Mazumder, Ali Saab",
     author_email="alikassemsaab@gmail.com",
     url='https://github.com/alisaab/l0bnb',
-    download_url="https://github.com/alisaab/l0bnb/archive/0.0.4.tar.gz",
-    install_requires=["numpy >= 1.18.2", "scipy >= 1.4.1", "numba >= 0.48.0",
-                      "matplotlib >= 3.2.1"],
-    version="0.0.4",
+    download_url="https://github.com/alisaab/l0bnb/archive/0.0.5.tar.gz",
+    install_requires=["numpy >= 1.18.2", "scipy >= 1.4.1", "numba >= 0.48.0"],
+    version="0.0.5",
     packages=find_packages(),
     include_package_data=True,
     license=LICENSE,