Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-06 07:18:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

dvc: avoid collecting stages twice

Browse Source 
Signed-off-by: Ruslan Kuprieiev <[email protected]>
This commit is contained in:
Ruslan Kuprieiev
2018-07-14 10:49:32 +03:00
parent c636581699
commit 794dcc2a58
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dvc/project.py
+5 -4
View File
@@ -227,19 +227,20 @@ class Project(object):
return [stage]
def reproduce(self, target, recursive=True, force=False):
stages = nx.get_node_attributes(self.graph(), 'stage')
G = self.graph()
stages = nx.get_node_attributes(G, 'stage')
node = os.path.relpath(os.path.abspath(target), self.root_dir)
if node not in stages:
raise NotDvcFileError(target)
if recursive:
return self._reproduce_stages(stages, node, force)
return self._reproduce_stages(G, stages, node, force)
return self._reproduce_stage(stages, node, force)
def _reproduce_stages(self, stages, node, force):
def _reproduce_stages(self, G, stages, node, force):
result = []
for n in nx.dfs_postorder_nodes(self.graph(), node):
for n in nx.dfs_postorder_nodes(G, node):
try:
result += self._reproduce_stage(stages, n, force)
except Exception as ex: