From cdf8523c94d98bea65bbf89f6e76a299188e9f19 Mon Sep 17 00:00:00 2001
From: Wojciech Lidwin <59863785+Halal37@users.noreply.github.com>
Date: Fri, 12 May 2023 04:39:40 +0200
Subject: [PATCH] Fix

---
 ium_sacred.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ium_sacred.py b/ium_sacred.py
index 43ed52b..9225467 100644
--- a/ium_sacred.py
+++ b/ium_sacred.py
@@ -53,7 +53,7 @@ def my_config():
 def prepare_message(epochs, lr, validation_split):
     return "{0} {1} {2}!".format(epochs, lr, validation_split)
 
-@ex.main
+@ex.automain
 def my_main(epochs, lr, validation_split, _run):
     train = pd.read_csv('baltimore_train.csv')