From b92b8a156dded88036b80079ed80867543836ab6 Mon Sep 17 00:00:00 2001
From: Louis Bethune <alguerythme@gmail.com>
Date: Thu, 14 Mar 2024 10:54:15 +0100
Subject: [PATCH] fix test action

---
 .github/workflows/tests.yml | 5 +++--
 requirements.txt            | 5 ++++-
 requirements_dev.txt        | 5 +----
 3 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 9c665bf..6b5cb7e 100644
--- a/.github/workflows/tests.yml
+++ b/.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
diff --git a/requirements.txt b/requirements.txt
index cc787fe..fa3f748 100644
--- a/requirements.txt
+++ b/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
\ No newline at end of file
diff --git a/requirements_dev.txt b/requirements_dev.txt
index ae50709..614f9c8 100644
--- a/requirements_dev.txt
+++ b/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
\ No newline at end of file
+bump2version
\ No newline at end of file