Fix: save model as .keras temp file

equinor · Jun 25, 2024 · 8f738fb · 8f738fb
1 parent 0150f0f
commit 8f738fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gordo/machine/model/models.py b/gordo/machine/model/models.py
@@ -179,7 +179,7 @@ def __getstate__(self):
 
         if hasattr(self, "model") and self.model is not None:
             with tempfile.NamedTemporaryFile("w", suffix=".keras") as tf:
-                save_model(self.model, tf, overwrite=True)
+                save_model(self.model, tf.name, overwrite=True)
                 tf.seek(0)
                 state["model"] = tf
             if hasattr(self, "history"):