From 4d8955dfd34986466914a832212d290c4b15bc4e Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Tue, 3 Sep 2019 12:59:32 +0300
Subject: [PATCH] Add package install options, pip install
 trains[gpu,s3,azure,gs]

---
 setup.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/setup.py b/setup.py
index deaa837c..8cb53d71 100644
--- a/setup.py
+++ b/setup.py
@@ -62,6 +62,20 @@ setup(
              'deeplearning deep-learning experiment-manager experimentmanager',
     packages=find_packages(exclude=['contrib', 'docs', 'data', 'examples', 'tests']),
     install_requires=requirements,
+    extras_require={
+        's3': [
+            'boto3>=1.9',
+        ],
+        'azure': [
+            'azure.storage.blob>=2.0.1',
+        ],
+        'gs': [
+            'google-cloud-storage>=1.13.2',
+        ],
+        'gpu': [
+            'gpustat',
+        ],
+    },
     package_data={
         'trains': ['config/default/*.conf', 'backend_api/config/default/*.conf']
     },