pin TF version so there is no memory overflow

arjunsavel · Mar 12, 2024 · 3b0aa74 · 3b0aa74
1 parent 823284e
commit 3b0aa74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -42,7 +42,7 @@ dynamic = ["version"]
 
 
 [project.optional-dependencies]
-neural_networks = ["tensorflow<2.16",
+neural_networks = ["tensorflow<=2.15.0",
                 "keras"]
 
 [project.urls]