From 360a3901d3fbb247597f5a068437a26b539db2bc 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:52:50 +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 3b05423..4417564 100644
--- a/train.py
+++ b/train.py
@@ -37,7 +37,7 @@ def my_main(EPOCHS, _run):
     norm_games_model.fit(games_all, games_predict, epochs=EPOCHS)
 
     norm_games_model.save('test')
-    print(f'done:')
+    _run.add_artifact('test')
 
 exint.run()
 exint.add_artifact("test")