Update setup.cfg

jezsadler · Apr 22, 2024 · 83f5215 · 83f5215
1 parent 3b19e4c
commit 83f5215
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -72,14 +72,14 @@ testing =
     nbmake
     tox
     flake8
-    tensorflow
+    tensorflow-cpu
     ipywidgets
     jupyter
     lightgbm
     linear-tree
     matplotlib
     pandas
-    keras
+    keras==2.9.0
     onnx
     onnxruntime
     onnxmltools