Fix dependency versions (#162)

* Fix dependency versions

* Update test dependencies

* Retrigger tests

* Retrigger tests

* Pin test dependency versions

* Pin test dependency versions

* Pin tox.ini dependencies

* Pause CPU integration tests

* Enable CPU test

---------

Co-authored-by: Ubuntu <[email protected]>

Author: sachanub

setup.py

@@ -52,8 +52,8 @@ def read(fname):
 
     # We don't declare our dependency on torch here because we build with
     # different packages for different variants
-    install_requires=['boto3==1.28.60', 'numpy==1.24.4', 'six==1.16.0',
-                      'retrying==1.3.4', 'scipy==1.10.1', 'psutil==5.9.5'],
+    install_requires=['boto3>=1.28.60', 'numpy>=1.24.4', 'six>=1.16.0',
+                      'retrying>=1.3.4', 'scipy>=1.10.1', 'psutil>=5.9.5'],
     extras_require={
         'test': ['coverage==7.3.2', 'docker-compose==1.29.2', 'flake8==6.1.0', 'Flask==3.0.0',
                  'mock==5.1.0', 'pytest==7.4.2', 'pytest-cov==4.1.0', 'pytest-xdist==3.3.1',

tox.ini

@@ -52,21 +52,32 @@ commands =
     {env:IGNORE_COVERAGE:} coverage report --fail-under=90
 
 deps =
-    coverage
-    pytest
-    pytest-cov
-    pytest-xdist
-    mock
-    requests
-    urllib3
+    coverage==7.3.2
+    filelock==3.12.3
+    botocore==1.31.67
+    typing_extensions==4.8.0
+    boto3==1.28.60
+    fsspec==2023.9.2
+    idna==3.4
+    psutil==5.9.5
+    charset-normalizer==3.3.0
+    pandas==2.0.3
+    networkx==3.1
+    schema==0.7.5
+    pytest==7.4.2
+    pytest-cov==4.1.0
+    pytest-xdist==3.3.1
+    mock==5.1.0
+    requests==2.31.0
+    urllib3==1.26.18
     sagemaker == 2.125.0
-    torch
-    torchvision
-    retrying
-    six
-    future
-    PyYaml
-    protobuf
+    torch==2.1.0
+    torchvision==0.16.0
+    retrying==1.3.4
+    six==1.16.0
+    future==0.18.3
+    PyYaml==6.0.1
+    protobuf==3.20.3
 
 [testenv:flake8]
 basepython = python3.8