Move dotTools module and test to pipe_base.

Author: TallJimbo

python/lsst/ctrl/mpexec/__init__.py

@@ -26,7 +26,6 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 from .cmdLineFwk import *
-from .dotTools import *
 from .executionGraphFixup import *
 from .mpGraphExecutor import *
 from .preExecInit import *

python/lsst/ctrl/mpexec/cmdLineFwk.py

@@ -68,14 +68,14 @@
     QuantumGraph,
     TaskFactory,
     buildExecutionButler,
+    graph2dot,
 )
 from lsst.pipe.base.all_dimensions_quantum_graph_builder import AllDimensionsQuantumGraphBuilder
 from lsst.pipe.base.pipeline_graph import NodeType
 from lsst.utils import doImportType
 from lsst.utils.logging import getLogger
 from lsst.utils.threads import disable_implicit_threading
 
-from .dotTools import graph2dot
 from .executionGraphFixup import ExecutionGraphFixup
 from .mpGraphExecutor import MPGraphExecutor
 from .preExecInit import PreExecInit, PreExecInitLimited
@@ -583,7 +583,10 @@ def makePipeline(self, args: SimpleNamespace) -> Pipeline:
             pipeline.write_to_uri(args.save_pipeline)
 
         if args.expand_pipeline:
-            pipeline.write_to_uri(args.expand_pipeline, expand=True)
+            task_defs = list(pipeline)
+            pipeline.write_to_uri(args.expand_pipeline, expand=True, task_defs=task_defs)
+        else:
+            task_defs = None
 
         return pipeline