From 05d8f7ca61cfe7759f056e297f468fcac22bd2c0 Mon Sep 17 00:00:00 2001
From: Andrzej Preibisz <andrzejjpreibisz@gmail.com>
Date: Mon, 6 Jun 2022 09:50:02 +0200
Subject: [PATCH] Yaml fix

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

diff --git a/dvc.yaml b/dvc.yaml
index 3e57527..56df93e 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -1,5 +1,11 @@
 stages:
   prepare:
-    cmd: '" -d" ml_prepare.py -o training_data.csv test_data.csv " python" ml_prepare.py'
+    cmd: python3 ml_prepare.py
+    deps:
+      - training_data.csv
+      - test_data.csv
+      - ml_prepare.py
   training:
-    cmd: '" -d" ml_training.py " -o" trained_model/ " python" ml_training.py 15'
+    cmd: python3 ml_training.py 15
+    deps:
+      - ml_training.py