diff --git a/examples/TensorFlow/Keras/IRIS/keras_to_MDF.py b/examples/TensorFlow/Keras/IRIS/keras_to_MDF.py index 1ee90ed2..96780eaf 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 1afce8ca..c0b0ea37 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 7718e0f9..1e30aa43 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