From 9f2924846346f8418dc5b232957e8656d773a48a Mon Sep 17 00:00:00 2001 From: David Turner Date: Wed, 20 Mar 2024 14:00:50 -0400 Subject: [PATCH] Re-exclude keras_model.py again. Actually, I don't like depending on test execution order. If we ever parallelize tests this will be bad. Lets try using exec. I think this will ensure the keras_model.py that is local to the script runs each and every time. --- examples/TensorFlow/Keras/IRIS/keras_to_MDF.py | 2 ++ examples/TensorFlow/Keras/MNIST/keras_to_MDF.py | 2 ++ tests/test_examples.py | 1 + 3 files changed, 5 insertions(+) diff --git a/examples/TensorFlow/Keras/IRIS/keras_to_MDF.py b/examples/TensorFlow/Keras/IRIS/keras_to_MDF.py index 1ee90ed24..96780eafc 100644 --- a/examples/TensorFlow/Keras/IRIS/keras_to_MDF.py +++ b/examples/TensorFlow/Keras/IRIS/keras_to_MDF.py @@ -11,6 +11,8 @@ from modelspec.utils import _val_info from modeci_mdf.execution_engine import EvaluableGraph +exec(open("./keras_model.py").read()) + # load the keras model model = tf.keras.models.load_model("keras_model_on_iris.keras") diff --git a/examples/TensorFlow/Keras/MNIST/keras_to_MDF.py b/examples/TensorFlow/Keras/MNIST/keras_to_MDF.py index 1afce8cae..c0b0ea37a 100644 --- a/examples/TensorFlow/Keras/MNIST/keras_to_MDF.py +++ b/examples/TensorFlow/Keras/MNIST/keras_to_MDF.py @@ -7,6 +7,8 @@ from modelspec.utils import _val_info from modeci_mdf.execution_engine import EvaluableGraph +exec(open("./keras_model.py").read()) + # load the keras model model = tf.keras.models.load_model("kr_N_model.keras") diff --git a/tests/test_examples.py b/tests/test_examples.py index 7718e0f9e..1e30aa436 100644 --- a/tests/test_examples.py +++ b/tests/test_examples.py @@ -28,6 +28,7 @@ ".reconstructed.py", "generate_json_and_scripts.py", "pytorch_ddm.py", + "keras_model.py", ] # Filter any excluded example scripts