Fix: remove save_format param from save_model in `KerasBaseEstima…

…tor.__getstate__`
equinor · Jun 24, 2024 · b93af57 · b93af57
1 parent c520dbb
commit b93af57
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:
             buf = io.BytesIO()
             with h5py.File(buf, compression="lzf", mode="w") as h5:
-                save_model(self.model, h5, overwrite=True)
+                self.model.save(h5, overwrite=True, save_format="h5")
                 buf.seek(0)
                 state["model"] = buf
             if hasattr(self, "history"):