Update setup.py

apache · Aug 26, 2021 · af04a0f · af04a0f
1 parent b73e60b
commit af04a0f
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/submarine-sdk/pysubmarine/setup.py b/submarine-sdk/pysubmarine/setup.py
@@ -33,18 +33,18 @@
         'sqlalchemy',
         'sqlparse',
         'pymysql',
-        'requests',
-        'urllib3 >= 1.15.1',
-        'certifi >= 14.05.14',
-        'python-dateutil >= 2.5.3',
+        'requests==2.26.0',
+        'urllib3>=1.15.1',
+        'certifi>=14.05.14',
+        'python-dateutil>=2.5.3',
         'pyarrow==0.17.0',
         'mlflow>=1.15.0',
         'boto3>=1.17.58',
     ],
     extras_require={
-        'tf':['tensorflow>=1.14.0,<2.0.0'],
-        'tf-latest':['tensorflow'],
-        'pytorch':['torch>=1.5.0','torchvision>=0.6.0'],
+        'tf': ['tensorflow>=1.14.0,<2.0.0'],
+        'tf-latest': ['tensorflow'],
+        'pytorch': ['torch>=1.5.0', 'torchvision>=0.6.0'],
     },
     classifiers=[
         'Intended Audience :: Developers',