XX

Signed-off-by: Matthew Ballance <[email protected]>
dv-flow · Jan 24, 2025 · ea66719 · ea66719
1 parent 01afa2a
commit ea66719
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/dv_flow/mgr/package_def.py b/src/dv_flow/mgr/package_def.py
@@ -58,7 +58,7 @@ def getTask(self, name : str) -> 'TaskDef':
     def mkPackage(self, session, params : Dict[str,Any] = None) -> 'Package':
         ret = Package(self.name)
 
-        session.push_package(ret)
+        session.push_package(ret, add=True)
 
         tasks_m : Dict[str,str,TaskCtor]= {}
 

diff --git a/src/dv_flow/mgr/task_graph_builder.py b/src/dv_flow/mgr/task_graph_builder.py
@@ -48,8 +48,10 @@ def __post_init__(self):
             self._pkg_spec_s.pop()
             self._pkg_m[PackageSpec(self.root_pkg.name)] = pkg
 
-    def push_package(self, pkg : Package):
+    def push_package(self, pkg : Package, add=False):
         self._pkg_s.append(pkg)
+        if add:
+            self._pkg_m[PackageSpec(pkg.name, pkg.params)] = pkg
 
     def pop_package(self, pkg : Package):
         self._pkg_s.pop()