Merge 991678c into 522cf75

Lightning-Universe · Feb 19, 2021 · 957ff0a · 957ff0a
2 parents 522cf75 + 991678c
commit 957ff0a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 11 deletions.
diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
@@ -71,7 +71,7 @@ jobs:
       uses: actions/cache@v2
       with:
         path: data # This path is specific to Ubuntu
-        key: flash-datasets-${{ hashFiles('tests/examples/test_examples.py') }}
+        key: flash-datasets-${{ hashFiles('tests/examples/test_scripts.py') }}
         restore-keys: flash-datasets-
 
     - name: Tests

diff --git a/requirements.txt b/requirements.txt
@@ -1,14 +1,13 @@
-pytorch-lightning==1.2.0rc0
-torch==1.7.1
-PyYAML==5.3.1
+pytorch-lightning==1.2.0rc0  # todo: we shall align with real 1.2
+torch>=1.7  # TODO: regenerate weights with lewer PT version
+PyYAML>=5.1
 Pillow>=7.2
-torchvision==0.8.2
-transformers==4.2.2
-pytorch-tabnet==3.1.1
-datasets==1.2.1
-pandas==1.1.5 ; python_version < "3.7"
-pandas==1.2.2 ; python_version >= "3.7"
-scikit-learn==0.24.0
+torchvision>=0.8  # lower to 0.7 after PT 1.6
+transformers>=4.0
+pytorch-tabnet==3.1
+datasets>=1.2, <1.3
+pandas>=1.1
+scikit-learn>=0.24
 numpy  # comes with 3rd-party dependency
 tqdm  # comes with 3rd-party dependency
 rouge-score>=0.0.4

diff --git a/tests/examples/test_examples.py → tests/examples/test_scripts.py b/tests/examples/test_examples.py → tests/examples/test_scripts.py