run tests with a later version of tensorflow

ThomasMBury · Feb 5, 2025 · 52c37e3 · 52c37e3
1 parent 7d9b814
commit 52c37e3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -26,7 +26,8 @@ jobs:
           python -m pip install --upgrade pip
           pip install flake8 pytest pytest-cov
           pip --default-timeout=100 install -e .
-          pip --default-timeout=100 install "tensorflow>=2.0.0,<2.12"
+          # pip --default-timeout=100 install "tensorflow>=2.0.0,<2.12"
+          pip --default-timeout=100 install "tensorflow>=2.14"
       - name: Lint with flake8
         run: |
           # stop the build if there are Python syntax errors or undefined names

diff --git a/tests/test_ewstools.py b/tests/test_ewstools.py
@@ -235,7 +235,10 @@ def test_TimeSeries_dl_preds():
     ts.detrend()
 
     # Import a classifier
-    classifier_path = "saved_classifiers/bury_pnas_21/len500/best_model_1_1_len500.pkl"
+    # classifier_path = "saved_classifiers/bury_pnas_21/len500/best_model_1_1_len500.pkl"
+    classifier_path = (
+        "saved_classifiers/bury_pnas_21/len500/best_model_1_1_len500.keras"
+    )
     classifier = load_model(classifier_path)
 
     # Apply classifier with time bounds
@@ -253,7 +256,10 @@ def test_TimeSeries_dl_preds():
     assert type(fig) == plotly.graph_objs._figure.Figure
 
     # Import and apply a second classifier
-    classifier_path = "saved_classifiers/bury_pnas_21/len500/best_model_1_2_len500.pkl"
+    # classifier_path = "saved_classifiers/bury_pnas_21/len500/best_model_1_2_len500.pkl"
+    classifier_path = (
+        "saved_classifiers/bury_pnas_21/len500/best_model_1_2_len500.keras"
+    )
     classifier2 = load_model(classifier_path)
     ts.apply_classifier_inc(classifier2, inc=40, name="c2", verbose=1)