Update generative-proof-of-concept-CPU-preprocessing-in-memory.py

david-thrower · web-flow · commit 14720e6b11ef · 2025-09-23T17:03:00.000-04:00
Change the purge_model_storage_files arg to the fit the updated API.
diff --git a/generative-proof-of-concept-CPU-preprocessing-in-memory.py b/generative-proof-of-concept-CPU-preprocessing-in-memory.py
@@ -730,7 +730,7 @@ def reset_state(self):
         
         MODEL_FILE_NAME = "cerebros-foundation-model.keras"
         
-        best_model_found = cerebros_automl.get_best_model(purge_model_storage_files=True)
+        best_model_found = cerebros_automl.get_best_model(purge_model_storage_files='slate')
         mlflow.keras.log_model(best_model_found, artifact_path="base")
         # best_model_found.save(MODEL_FILE_NAME)
         # del(best_model_found)
