Merge pull request #1367 from JosepSampe/lithops-dev

[Setup] Move matplotlib, seaborn and pandas to 'plotting' extra

Author: JosepSampe

lithops/executors.py

@@ -543,8 +543,13 @@ def plot(
             logger.debug(f'ExecutorID {self.executor_id} - No futures ready to plot')
             return
 
-        logging.getLogger('matplotlib').setLevel(logging.WARNING)
-        from lithops.plots import create_timeline, create_histogram
+        try:
+            logging.getLogger('matplotlib').setLevel(logging.WARNING)
+            from lithops.plots import create_timeline, create_histogram
+        except ImportError:
+            raise ModuleNotFoundError(
+                "Please install 'pip3 install lithops[plotting]' for "
+                "making use of the plot() method")
 
         logger.info(f'ExecutorID {self.executor_id} - Creating execution plots')
 
@@ -628,8 +633,13 @@ def job_summary(self, cloud_objects_n: Optional[int] = 0):
 
         :param cloud_objects_n: number of cloud object used in COS, declared by user.
         """
-        import pandas as pd
-        import numpy as np
+        try:
+            import pandas as pd
+            import numpy as np
+        except ImportError:
+            raise ModuleNotFoundError(
+                "Please install 'pip3 install lithops[plotting]' for "
+                "making use of the job_summary() method")
 
         def init():
             headers = ['Job_ID', 'Function', 'Invocations', 'Memory(MB)', 'AvgRuntime', 'Cost', 'CloudObjects']

setup.py

@@ -6,15 +6,13 @@
 install_requires = [
     'Click',
     'tabulate',
-    'pandas',
+    'six',
     'PyYAML',
     'pika',
     'tqdm',
     'tblib',
     'requests',
-    'seaborn',
     'paramiko',
-    'matplotlib',
     'cloudpickle',
     'tblib',
     'ps-mem',
@@ -23,20 +21,22 @@
 
 
 extras_require = {
+    'ibm': [
+        'ibm-cos-sdk',
+        'ibm-code-engine-sdk',
+        'ibm-vpc',
+        'kubernetes',
+    ],
+    'aws': [
+        'boto3'
+    ],
     'gcp': [
         'httplib2',
         'google-cloud-storage',
         'google-cloud-pubsub',
         'google-api-python-client',
         'google-auth'
     ],
-    'aliyun': [
-        'aliyun-fc2',
-        'oss2'
-    ],
-    'aws': [
-        'boto3'
-    ],
     'azure': [
         'azure-mgmt-resource',
         'azure-mgmt-compute',
@@ -45,15 +45,13 @@
         'azure-storage-blob',
         'azure-storage-queue'
     ],
+    'aliyun': [
+        'aliyun-fc2',
+        'oss2'
+    ],
     'ceph': [
         'boto3'
     ],
-    'ibm': [
-        'ibm-cos-sdk',
-        'ibm-code-engine-sdk',
-        'ibm-vpc',
-        'kubernetes',
-    ],
     'knative': [
         'kubernetes',
     ],
@@ -75,6 +73,12 @@
         'diskcache',
         'numpy'
     ],
+    'plotting': [
+        'pandas',
+        'matplotlib',
+        'seaborn',
+        'numpy'
+    ],
     'oracle': [
         'oci',
     ],
@@ -110,9 +114,6 @@
         'Operating System :: OS Independent',
         'Natural Language :: English',
         'License :: OSI Approved :: Apache Software License',
-        'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.6',
-        'Programming Language :: Python :: 3.7',
         'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',
         'Programming Language :: Python :: 3.10',