Revert tensorflow bump up back to 2.15.1

DanielTsiang · May 1, 2024 · f68f59b · f68f59b
1 parent c8e3f99
commit f68f59b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/services/flask/requirements.txt b/services/flask/requirements.txt
@@ -10,5 +10,4 @@ pytest~=8.2.0
 requests~=2.31.0
 setuptools>=65.5.1 # not directly required, pinned to avoid a vulnerability
 # tensorflow is a testing requirement
-tensorflow~=2.16.1
-tf-keras~=2.16
+tensorflow~=2.15.1
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -4,15 +4,15 @@
 import cv2
 import numpy as np
 import pytest
-import tf_keras as keras
+import tensorflow as tf
 from conftest import *
 
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"  # Disable tensorflow debugging logs
 
 
 @pytest.fixture(scope="module")
 def model():
-    return keras.models.load_model(
+    return tf.keras.models.load_model(
         Path(__file__).parents[1].resolve()
         / "services"
         / "tensorflow-serving"