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

---
 Jenkinsfile_train | 2 +-
 ium_sacred.py     | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index d94ffab..b8baddf 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -25,7 +25,7 @@ node {
      checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's487197', url: 'https://git.wmi.amu.edu.pl/s487197/ium_487197']]])
  }
  stage('Dockerfile'){
-    def testImage = docker.image('s487197/ium:45')
+    def testImage = docker.image('s487197/ium:46')
 	testImage.inside{ 
          copyArtifacts filter: 'baltimore_train.csv', projectName: 's487197-create-dataset'
 		 sh "python3 ium_sacred.py -epochs $EPOCHS -lr $LR -validation_split $VALIDATION_SPLIT"
diff --git a/ium_sacred.py b/ium_sacred.py
index 109cfa7..ff18e32 100644
--- a/ium_sacred.py
+++ b/ium_sacred.py
@@ -14,7 +14,8 @@ import math
 
 ex = Experiment('s487197-train', save_git_info=False,interactive=True)
 
-
+ex.observers.append(FileStorageObserver('sacred_results'))
+ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', db_name='sacred'))
 #ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', db_name='sacred'))
 
 
@@ -79,8 +80,7 @@ def my_main(epochs, lr, validation_split, _run):
         _run.log_scalar('training.loss', his[1]['loss'])
         _run.log_scalar('accuracy', his[1]['accuracy'])
     ex.add_artifact('baltimore_model')
-    ex.observers.append(FileStorageObserver('sacred_results'))
-    ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', db_name='sacred'))
+
 """
     baltimore_data_test =pd.read_csv('baltimore_test.csv')
     baltimore_data_test.columns = train.columns