From dc0adbffd9a179163c92a9a0df5d338adcba7373 Mon Sep 17 00:00:00 2001 From: Caparrini Date: Sun, 3 Mar 2024 10:11:23 +0100 Subject: [PATCH] Modified: test for optional keras, added installation of tensorflow --- mloptimizer/test/test_alg_wrapper.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mloptimizer/test/test_alg_wrapper.py b/mloptimizer/test/test_alg_wrapper.py index d96e09b..9b3e18f 100644 --- 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,