Skip to content

Commit

Permalink
Fixup
Browse files Browse the repository at this point in the history
Signed-off-by: Matthew Ballance <[email protected]>
  • Loading branch information
mballance committed Jan 22, 2025
1 parent 5edb3c5 commit 373da6c
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 6 deletions.
11 changes: 6 additions & 5 deletions src/dv_flow/mgr/task_graph_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,11 +22,11 @@ def __post_init__(self):
if self.pkg_rgy is None:
self.pkg_rgy = PkgRgy.inst()

self._pkg_spec_s.append(self.root_pkg)
pkg = self.root_pkg.mkPackage(self)
self._pkg_spec_s.pop()
self._pkg_m[PackageSpec(self.root_pkg.name)] = pkg
print("pkg_m: %s" % self.root_pkg.name)
if self.root_pkg is not None:
self._pkg_spec_s.append(self.root_pkg)
pkg = self.root_pkg.mkPackage(self)
self._pkg_spec_s.pop()
self._pkg_m[PackageSpec(self.root_pkg.name)] = pkg

def push_package(self, pkg : Package):
self._pkg_s.append(pkg)
Expand Down Expand Up @@ -157,6 +157,7 @@ def getPackage(self, spec : PackageSpec) -> Package:
break

if pkg is None:
print("Checking registry")
p_def = self.pkg_rgy.getPackage(spec.name)

if p_def is not None:
Expand Down
2 changes: 1 addition & 1 deletion src/dv_flow/mgr/task_graph_runner_local.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ def __post_init__(self):


async def exec(self, *args, **kwargs):
return self.create_subprocess(*args, **kwargs)
return await self.create_subprocess(*args, **kwargs)

async def run(self, task : Union[Task,List[Task]]) -> List['TaskData']:
if isinstance(task, Task):
Expand Down

0 comments on commit 373da6c

Please sign in to comment.