refactor: extract stage loading outside of dvcfile (#3655)

* refactor: extract stage loading outside of dvcfile * repo: simplify stage collection * fix typing
iterative · Apr 20, 2020 · 63c466d · 63c466d
1 parent 224b0fc
commit 63c466d
Show file tree

Hide file tree

Showing 24 changed files with 404 additions and 377 deletions.
diff --git a/dvc/command/pipeline.py b/dvc/command/pipeline.py
@@ -25,7 +25,7 @@ def _show(self, target, commands, outs, locked):
         from dvc.utils import parse_target
 
         path, name = parse_target(target)
-        stage = dvcfile.Dvcfile(self.repo, path).load_one(name)
+        stage = dvcfile.Dvcfile(self.repo, path).stages[name]
         G = self.repo.pipeline_graph
         stages = networkx.dfs_postorder_nodes(G, stage)
         if locked:
@@ -49,7 +49,7 @@ def _build_graph(self, target, commands=False, outs=False):
         from dvc.utils import parse_target
 
         path, name = parse_target(target)
-        target_stage = dvcfile.Dvcfile(self.repo, path).load_one(name)
+        target_stage = dvcfile.Dvcfile(self.repo, path).stages[name]
         G = get_pipeline(self.repo.pipelines, target_stage)
 
         nodes = set()