Yaml fix

2022-06-06 09:50:02 +02:00 · 2022-06-06 09:50:02 +02:00 · 05d8f7ca61
commit 05d8f7ca61
parent 108b776f22
1 changed files with 8 additions and 2 deletions
--- 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