Skip to content

Commit

Permalink
Un-exclude keras_model.
Browse files Browse the repository at this point in the history
Ok, I think that keras_model.py is not running twice in some
cases because the module with the same name is getting cached.
I have re-included it and it seems that it should always run
first anyway.
  • Loading branch information
davidt0x committed Mar 20, 2024
1 parent 290cd8c commit 281ca4e
Show file tree
Hide file tree
Showing 3 changed files with 0 additions and 9 deletions.
4 changes: 0 additions & 4 deletions examples/TensorFlow/Keras/IRIS/keras_to_MDF.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,6 @@
from modelspec.utils import _val_info
from modeci_mdf.execution_engine import EvaluableGraph

# Import keras_model to ensure it runs and generates the model
sys.path.append(".")
import keras_model # noqa: F401

# load the keras model
model = tf.keras.models.load_model("keras_model_on_iris.keras")

Expand Down
4 changes: 0 additions & 4 deletions examples/TensorFlow/Keras/MNIST/keras_to_MDF.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,6 @@
from modelspec.utils import _val_info
from modeci_mdf.execution_engine import EvaluableGraph

# Import keras_model to ensure it runs and generates the model
sys.path.append(".")
import keras_model # noqa: F401

# load the keras model
model = tf.keras.models.load_model("kr_N_model.keras")

Expand Down
1 change: 0 additions & 1 deletion tests/test_examples.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,6 @@
".reconstructed.py",
"generate_json_and_scripts.py",
"pytorch_ddm.py",
"keras_model.py",
]

# Filter any excluded example scripts
Expand Down

0 comments on commit 281ca4e

Please sign in to comment.