fix test action

.github/workflows/tests.yml

@@ -27,18 +27,19 @@ jobs:
           set -xe
           pip install --upgrade pip setuptools wheel
           pip install -r requirements.txt
+          pip install -r requirements_dev.txt
         shell: bash
       - name: Build
         run: |
           set -xe
           python -VV
-          python setup.py install
+          python -m pip install .
         shell: bash
       - name: Run tests
         timeout-minutes: 60
         run: |
           set -xe
           python -VV
-          python -c "import tf; print('tf', tf.__version__)"
+          python -c "import tensorflow as tf; print('tf', tf.__version__)"
           pytest tests
         shell: bash

requirements.txt

@@ -1,8 +1,11 @@
 scipy<=1.9.3
 autodp
+absl-py
 numpy
 deel-lip
 matplotlib
+ml_collections
 pandas
-tensorflow
+tensorflow<2.16
 tensorflow-datasets
+wandb

requirements_dev.txt

@@ -1,7 +1,5 @@
 setuptools
 pre-commit
-ml_collections
-absl-py
 pytest
 tox
 black
@@ -11,5 +9,4 @@ mkdocs
 mkdocs-material
 mkdocstrings[python]
 mknotebooks
-bump2version
-wandb
+bump2version