diff --git a/setup.cfg b/setup.cfg
index 7b4233e8..7ea8599e 100644
--- 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