From 65b73dc86feddf253f0c13c8674c8746c4ece3ed Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= <patgal2@st.amu.edu.pl>
Date: Thu, 11 May 2023 21:36:27 +0200
Subject: [PATCH] Update 'train.py'

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

diff --git a/train.py b/train.py
index 21dd0cc..bf2394d 100644
--- a/train.py
+++ b/train.py
@@ -17,7 +17,7 @@ def my_config():
     
 
 @exint.main
-def my_main(EPOCHS):
+def my_main(EPOCHS, _run):
     _run.info["epochs"] = EPOCHS
     normalize = layers.Normalization()
     train_data_x = exint.open_resource('./X_train.csv', 'r')