From 4ad41fe07273d22c6bc86d2658bca3387666b536 Mon Sep 17 00:00:00 2001
From: Agata Kulhawczuk <agakul1@st.amu.edu.pl>
Date: Mon, 30 May 2022 21:52:31 +0200
Subject: [PATCH] Zaktualizuj 'dvc.yaml'

---
 dvc.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dvc.yaml b/dvc.yaml
index cfce856..9c05247 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -1,6 +1,6 @@
 stages:
   prepare_datasets:
-    cmd: python prepare_datasets.py X_train.csv X_test.csv y_train.csv y_test.csv
+    cmd: python prepare_datasets.py
     deps:
     - data/bodyPerformance.csv
     - prepare_datasets.py
@@ -10,7 +10,7 @@ stages:
     - y_test.csv
     - y_train.csv
   train:
-    cmd: python train.py model.pth
+    cmd: python train.py
     deps:
     - X_train.csv
     - train.py