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

---
 dvc.yaml | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/dvc.yaml b/dvc.yaml
index 2c85c35..fbf0d84 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -1,11 +1,10 @@
 stages:
   prepare_datasets:
-    cmd: python3 prepare_datasets.py X_train.csv X_test.csv y_train.csv y_test.csv
+    cmd: python3 prepare_datasets.py 
     deps:
     - data/bodyPerformance.csv
     - prepare_datasets.py
-    outs:
-    - X_test.csv
-    - X_train.csv
-    - y_test.csv
-    - y_train.csv
+  train:
+    cmd: python train.py
+    deps:
+    - train.py