Move dotTools module and test to pipe_base.

Author: TallJimbo

python/lsst/ctrl/mpexec/__init__.py

@@ -20,7 +20,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

@@ -53,11 +53,12 @@
     QuantumGraph,
     TaskDef,
     buildExecutionButler,
+    graph2dot,
+    pipeline2dot,
 )
 from lsst.utils import doImport
 
 from . import util
-from .dotTools import graph2dot, pipeline2dot
 from .executionGraphFixup import ExecutionGraphFixup
 from .mpGraphExecutor import MPGraphExecutor
 from .preExecInit import PreExecInit
@@ -534,10 +535,15 @@ def makePipeline(self, args):
             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
 
         if args.pipeline_dot:
-            pipeline2dot(pipeline, args.pipeline_dot)
+            # Pass in existing task_defs if we have them to avoid re-importing
+            # and expanding everything.
+            pipeline2dot(pipeline if task_defs is None else task_defs, args.pipeline_dot)
 
         return pipeline