Merge pull request #240 from SyntenyBio/jday1/support-tensorflow217

Remove upper bound on tensorflow
openvax · Sep 18, 2024 · 75251b8 · 75251b8
2 parents f817c51 + e88ba1c
commit 75251b8
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
@@ -662,7 +662,6 @@ def wrapped_generator():
                 initial_epoch=epoch - 1,
                 epochs=epoch,
                 use_multiprocessing=False,
-                workers=0,
                 validation_data=(validation_x_dict, validation_y_dict),
                 verbose=verbose,
             )

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 six
 pandas>=0.20.3
-tensorflow>=2.15.0,<2.17.0
+tensorflow>=2.15.0
 tf-keras
 appdirs
 scikit-learn

diff --git a/setup.py b/setup.py
@@ -41,7 +41,7 @@
         "mhcgnomes>=0.8.4",
         "pyyaml",
         "tqdm",
-        "tensorflow>=2.15.0,<2.17.0",
+        "tensorflow>=2.15.0",
         "tf-keras"
     ]