From aa20f3caa49c9583702262c769a3a8e85b496d1d Mon Sep 17 00:00:00 2001
From: Witold Woch <witwoc@st.amu.edu.pl>
Date: Sun, 14 May 2023 18:13:25 +0200
Subject: [PATCH] mlflow fix

---
 MLproject.yml | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/MLproject.yml b/MLproject.yml
index 7a7b3c5..e9e0b44 100644
--- a/MLproject.yml
+++ b/MLproject.yml
@@ -4,13 +4,12 @@ docker_env:
   image: s487194/ium:1
 
 entry_points:
-  main:
+  train:
     parameters:
       epochs: {type: int, default: 100}
       lr: {type: float, default: 0.01}
     command: "python train.py --epochs {epochs} --lr {lr}"
+    
   test:
-    parameters:
-      
     command: "python prediction.py"