Modified: test for optional keras, added installation of tensorflow

Caparrini · Mar 3, 2024 · dc0adbf · dc0adbf
1 parent 5aab2c9
commit dc0adbf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mloptimizer/test/test_alg_wrapper.py b/mloptimizer/test/test_alg_wrapper.py
@@ -11,6 +11,7 @@ def test_keras_classifier():
     except ImportError as e:
         print(f"{e}: Keras is not installed. It will be installed to test.")
         subprocess.check_call([sys.executable, "-m", "pip", "install", "keras==2.12.0"])
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "tensorflow==2.12.1"])
         from keras.wrappers.scikit_learn import KerasClassifier
 
     uat = KerasClassifier(build_fn=generate_model, epochs=10,