diff --git a/dvc.yaml b/dvc.yaml
index 4ac5e1c..fb804a4 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -6,3 +6,5 @@ stages:
     - prepare_datasets.py
   train:
     cmd: python train.py
+    deps:
+    - train.py