sacred fix 1
Some checks failed
s464980-training/pipeline/head There was a failure building this commit

This commit is contained in:
Sheaza 2024-06-09 18:05:59 +02:00
parent 7a3e24a150
commit 343ebbf307

View File

@ -54,7 +54,7 @@ class RegressionModel:
def save_model(self): def save_model(self):
self.model.save("model.keras") self.model.save("model.keras")
ex.add_artifact("model.keras") ex.add_artifact("model.keras", content_type="application/octet-stream")
@ex.main @ex.main