lsst-pipe-base 29.2025.3100__py3-none-any.whl → 29.2025.3200__py3-none-any.whl

lsst/pipe/base/single_quantum_executor.py

@@ -66,12 +66,12 @@ class SingleQuantumExecutor(QuantumExecutor):
 
     Parameters
     ----------
-    butler : `~lsst.daf.butler.Butler` or `None`
-        Data butler, `None` means that Quantum-backed butler should be used
-        instead.
-    task_factory : `.TaskFactory`
-        Instance of a task factory.
-    skip_existing_in : `~typing.Any`
+    butler : `~lsst.daf.butler.Butler` or `None`, optional
+        Data butler, `None` means that a limited butler should be used instead.
+    task_factory : `.TaskFactory`, optional
+        Instance of a task factory.  Defaults to a new instance of
+        `lsst.pipe.base.TaskFactory`.
+    skip_existing_in : `str` or `~collections.abc.Iterable` [ `str` ]
         Expressions representing the collections to search for existing output
         datasets. See :ref:`daf_butler_ordered_collection_searches` for allowed
         types. This class only checks for the presence of butler output run in
@@ -117,8 +117,8 @@ class SingleQuantumExecutor(QuantumExecutor):
     def __init__(
         self,
         *,
-        butler: Butler | None,
-        task_factory: TaskFactory,
+        butler: Butler | None = None,
+        task_factory: TaskFactory | None = None,
         skip_existing_in: Any = None,
         clobber_outputs: bool = False,
         enable_lsst_debug: bool = False,
@@ -130,7 +130,7 @@ class SingleQuantumExecutor(QuantumExecutor):
         job_metadata: Mapping[str, int | str | float] | None = None,
     ):
         self._butler = butler
-        self._task_factory = task_factory
+        self._task_factory = task_factory if task_factory is not None else TaskFactory()
         self._clobber_outputs = clobber_outputs
         self._enable_lsst_debug = enable_lsst_debug
         self._limited_butler_factory = limited_butler_factory
@@ -171,8 +171,7 @@ class SingleQuantumExecutor(QuantumExecutor):
         """
         startTime = time.time()
 
-        # Make a limited butler instance if needed (which should be QBB if full
-        # butler is not defined).
+        # Make a limited butler instance if needed.
         limited_butler: LimitedButler
         if self._butler is not None:
             limited_butler = self._butler

lsst/pipe/base/tests/in_memory_limited_butler.py

@@ -0,0 +1,223 @@
+# This file is part of pipe_base.
+#
+# Developed for the LSST Data Management System.
+# This product includes software developed by the LSST Project
+# (http://www.lsst.org).
+# See the COPYRIGHT file at the top-level directory of this distribution
+# for details of code ownership.
+#
+# This software is dual licensed under the GNU General Public License and also
+# under a 3-clause BSD license. Recipients may choose which of these licenses
+# to use; please see the files gpl-3.0.txt and/or bsd_license.txt,
+# respectively.  If you choose the GPL option then the following text applies
+# (but note that there is still no warranty even if you opt for BSD instead):
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from __future__ import annotations
+
+__all__ = ["InMemoryLimitedButler"]
+
+import logging
+import uuid
+from collections.abc import Iterable
+from typing import Any
+
+from lsst.daf.butler import (
+    ButlerMetrics,
+    DatasetProvenance,
+    DatasetRef,
+    DatasetType,
+    DimensionUniverse,
+    LimitedButler,
+    MissingDatasetTypeError,
+    Quantum,
+    StorageClass,
+    StorageClassFactory,
+)
+from lsst.daf.butler.registry import ConflictingDefinitionError
+
+from .._dataset_handle import InMemoryDatasetHandle
+
+_LOG = logging.getLogger(__name__)
+
+
+class InMemoryLimitedButler(LimitedButler):
+    """A `LimitedButler` that just stores datasets in an in-memory mapping.
+
+    Parameters
+    ----------
+    universe : `lsst.daf.butler.DimensionUniverse`
+        Definitions for all dimensions.
+    dataset_types : `~collections.abc.Iterable` [ \
+            `lsst.daf.butler.DatasetType` ]
+        Definitions of all dataset types.
+
+    Notes
+    -----
+    This is an incomplete implementation of the `LimitedButler` interface
+    intended only for tests.  It supports all methods required by
+    `SingleQuantumExecutor`, but not transfers or URI retrieval.
+
+    While this class supports storage class conversions in `get` and `put`, it
+    uses different code paths from real butlers, and should not be used in
+    tests in which storage class correctness is part of what is being tested.
+
+    Objects are always copied (via storage class machinery) by `get`.
+
+    Pickling this class will pickle all datasets already `put` (which must be
+    pickleable).  This generally allows a central butler to be initialized with
+    input datasets in one process and distributed to worker processes that run
+    quanta *once*, but it does not allow outputs from a worker process to be
+    distributed to others or the originating process.  This can be hard to
+    notice because quanta will usually be skipped with
+    `lsst.pipe.base.NoWorkFound` (a success!) when all of their inputs are
+    missing.
+    """
+
+    def __init__(self, universe: DimensionUniverse, dataset_types: Iterable[DatasetType] = ()):
+        self.storageClasses = StorageClassFactory()
+        self._universe = universe
+        self._datasets: dict[uuid.UUID, tuple[DatasetRef, InMemoryDatasetHandle]] = {}
+        self._metrics = ButlerMetrics()
+        self._dataset_types = {dt.name: dt for dt in dataset_types}
+        assert not any(dt.component() for dt in self._dataset_types.values()), (
+            "Dataset type definitions must not be components."
+        )
+
+    def __getstate__(self) -> dict[str, Any]:
+        # Pickle customization is needed because StorageClassFactory is not
+        # pickleable.
+        return {
+            "universe": self._universe,
+            "datasets": self._datasets,
+            "dataset_types": self._dataset_types,
+        }
+
+    def __setstate__(self, state: dict[str, Any]) -> None:
+        self.storageClasses = StorageClassFactory()
+        self._universe = state["universe"]
+        self._datasets = state["datasets"]
+        self._metrics = ButlerMetrics()
+        self._dataset_types = state["dataset_types"]
+
+    def get_datasets(self, dataset_type: str | None = None) -> dict[DatasetRef, object]:
+        """Return datasets that have been `put` to this butler.
+
+        Storage classes and corresponding Python types will match the dataset
+        type definitions provided at butler construction, which may not be the
+        same as what was `put`.
+
+        Parameters
+        ----------
+        dataset_type : `str`, optional
+            Dataset type name used to filter results.
+
+        Returns
+        -------
+        refs : `dict` [ `lsst.daf.butler.DatasetRef`, `object` ]
+            Datasets held by this butler.
+        """
+        return {
+            ref: handle.get()
+            for ref, handle in self._datasets.values()
+            if dataset_type is None or dataset_type == ref.datasetType.name
+        }
+
+    def isWriteable(self) -> bool:
+        return True
+
+    def put(self, obj: Any, ref: DatasetRef, /, *, provenance: DatasetProvenance | None = None) -> DatasetRef:
+        with self._metrics.instrument_put():
+            assert not ref.isComponent(), "Component dataset types cannot be put."
+            if ref.id in self._datasets:
+                # Some butlers may not raise reliably when a dataset already
+                # exists (it's hard to be rigorous in parallel given different
+                # guarantees provided by storage), but we don't want code to
+                # rely on it not being an error, so we want a test butler to
+                # always complain.
+                raise ConflictingDefinitionError(f"Dataset {ref} already exists.")
+            if (repo_dataset_type := self._dataset_types.get(ref.datasetType.name)) is None:
+                raise MissingDatasetTypeError(f"Dataset type {ref.datasetType.name!r} not recognized.")
+            repo_dataset_type.storageClass.coerce_type(obj)
+            self._datasets[ref.id] = (
+                ref.overrideStorageClass(repo_dataset_type.storageClass),
+                InMemoryDatasetHandle(
+                    obj,
+                    storageClass=repo_dataset_type.storageClass,
+                    dataId=ref.dataId,
+                    copy=True,
+                ),
+            )
+        return ref
+
+    def get(
+        self,
+        ref: DatasetRef,
+        /,
+        *,
+        parameters: dict[str, Any] | None = None,
+        storageClass: StorageClass | str | None = None,
+    ) -> Any:
+        with self._metrics.instrument_get():
+            if storageClass is None:
+                storageClass = ref.datasetType.storageClass
+            elif isinstance(storageClass, str):
+                storageClass = self.storageClasses.getStorageClass(storageClass)
+            if entry := self._datasets.get(ref.id):
+                (ref, handle) = entry
+                return handle.get(
+                    component=ref.datasetType.component(), parameters=parameters, storageClass=storageClass
+                )
+        raise FileNotFoundError(f"Dataset {ref} does not exist.")
+
+    def stored_many(self, refs: Iterable[DatasetRef]) -> dict[DatasetRef, bool]:
+        return {ref: ref.id in self._datasets for ref in refs}
+
+    def pruneDatasets(
+        self,
+        refs: Iterable[DatasetRef],
+        *,
+        disassociate: bool = True,
+        unstore: bool = False,
+        tags: Iterable[str] = (),
+        purge: bool = False,
+    ) -> None:
+        for ref in refs:
+            self._datasets.pop(ref.id, None)
+
+    @property
+    def _datastore(self) -> Any:
+        raise NotImplementedError("This test butler does not have a datastore.")
+
+    @_datastore.setter  # demanded by MyPy since we declare it to be an instance attribute in LimitedButler.
+    def _datastore(self, value: Any) -> None:
+        raise NotImplementedError("This test butler does not have a datastore.")
+
+    @property
+    def dimensions(self) -> DimensionUniverse:
+        return self._universe
+
+    def factory(self, quantum: Quantum) -> InMemoryLimitedButler:
+        """Return ``self``.
+
+        This method can be used as the ``limited_butler_factory`` argument to
+        `.single_quantum_executor.SingleQuantumExecutor`.
+
+        Parameters
+        ----------
+        quantum : `lsst.daf.butler.Quantum`
+            Ignored.
+        """
+        return self

lsst/pipe/base/tests/mocks/__init__.py

@@ -32,5 +32,6 @@ See :ref:`testing-pipelines-with-mocks` for details.
 """
 
 from ._data_id_match import *
+from ._in_memory_repo import *
 from ._pipeline_task import *
 from ._storage_class import *

lsst/pipe/base/tests/mocks/_in_memory_repo.py

@@ -0,0 +1,357 @@
+# This file is part of pipe_base.
+#
+# Developed for the LSST Data Management System.
+# This product includes software developed by the LSST Project
+# (http://www.lsst.org).
+# See the COPYRIGHT file at the top-level directory of this distribution
+# for details of code ownership.
+#
+# This software is dual licensed under the GNU General Public License and also
+# under a 3-clause BSD license. Recipients may choose which of these licenses
+# to use; please see the files gpl-3.0.txt and/or bsd_license.txt,
+# respectively.  If you choose the GPL option then the following text applies
+# (but note that there is still no warranty even if you opt for BSD instead):
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from __future__ import annotations
+
+__all__ = ("InMemoryRepo",)
+
+from collections.abc import Iterable, Mapping
+from typing import Any
+
+from lsst.daf.butler import CollectionType, DataCoordinate, DatasetRef, DatasetType, RegistryConfig
+from lsst.daf.butler.tests.utils import create_populated_sqlite_registry
+from lsst.resources import ResourcePath, ResourcePathExpression
+from lsst.sphgeom import RangeSet
+
+from ...all_dimensions_quantum_graph_builder import AllDimensionsQuantumGraphBuilder
+from ...pipeline_graph import PipelineGraph
+from ...single_quantum_executor import SingleQuantumExecutor
+from ..in_memory_limited_butler import InMemoryLimitedButler
+from ._pipeline_task import (
+    DynamicConnectionConfig,
+    DynamicTestPipelineTask,
+    DynamicTestPipelineTaskConfig,
+)
+from ._storage_class import MockDataset, is_mock_name
+
+
+class InMemoryRepo:
+    """A test helper that simulates a butler repository for task execution
+    without any disk I/O.
+
+    Parameters
+    ----------
+    *args : `str` or `lsst.resources.ResourcePath`
+        Butler YAML import files to load into the test repository.
+    registry_config : `lsst.daf.butler.RegistryConfig`, optional
+        Registry configuration for the repository.
+    input_run : `str`, optional
+        Name of a `~lsst.daf.butler.CollectionType.RUN` collection that will be
+        used as an input to quantum graph generation.  Input datasets created
+        by the helper are added to this collection.
+    input_chain : `str`, optional
+        Name of a `~lsst.daf.butler.CollectionType.CHAINED` collection that
+        will be the direct input to quantum graph generation.  This always
+        includes ``input_run``.
+    output_run : `str`, optional
+        Name of a `~lsst.daf.butler.CollectionType.RUN` collection for
+        execution outputs.
+    use_import_collections_as_input : `bool` `str`, or \
+            `~collections.abc.Iterable` [ `str`], optional
+        Additional collections from YAML import files to include in
+        ``input_chain``, or `True` to include all such collections (in
+        chain-flattened lexicographical order).
+    data_root : convertible to `lsst.resources.ResourcePath`, optional
+        Root directory to join to each element in ``*args``.  Defaults to
+        the `lsst.daf.butler.tests.registry_data` package.
+
+    Notes
+    -----
+    This helper maintains an `..pipeline_graph.PipelineGraph` and a
+    no-datastore butler backed by an in-memory SQLite database for use in
+    quantum graph generation.
+    """
+
+    def __init__(
+        self,
+        *args: str | ResourcePath,
+        registry_config: RegistryConfig | None = None,
+        input_run: str = "input_run",
+        input_chain: str = "input_chain",
+        output_run: str = "output_run",
+        use_import_collections_as_input: bool | str | Iterable[str] = True,
+        data_root: ResourcePathExpression | None = "resource://lsst.daf.butler/tests/registry_data",
+    ):
+        if data_root is not None:
+            data_root = ResourcePath(data_root, forceDirectory=True)
+            args = tuple(data_root.join(arg) for arg in args)
+        self.butler = create_populated_sqlite_registry(*args, registry_config=registry_config)
+        input_chain_definition = [input_run]
+        if use_import_collections_as_input:
+            if use_import_collections_as_input is True:
+                use_import_collections_as_input = sorted(
+                    self.butler.collections.query("*", flatten_chains=True)
+                )
+            input_chain_definition += list(use_import_collections_as_input)
+        self.input_run = input_run
+        self.input_chain = input_chain
+        self.output_run = output_run
+        self.butler.collections.register(self.input_run)
+        self.butler.collections.register(self.input_chain, CollectionType.CHAINED)
+        self.butler.collections.redefine_chain(self.input_chain, input_chain_definition)
+        self.pipeline_graph = PipelineGraph()
+        self.last_auto_dataset_type_index = 0
+        self.last_auto_task_index = 0
+
+    def add_task(
+        self,
+        label: str | None = None,
+        *,
+        task_class: type[DynamicTestPipelineTask] = DynamicTestPipelineTask,
+        config: DynamicTestPipelineTaskConfig | None = None,
+        dimensions: Iterable[str] | None = None,
+        inputs: Mapping[str, DynamicConnectionConfig] | None = None,
+        outputs: Mapping[str, DynamicConnectionConfig] | None = None,
+        prerequisite_inputs: Mapping[str, DynamicConnectionConfig] | None = None,
+        init_inputs: Mapping[str, DynamicConnectionConfig] | None = None,
+        init_outputs: Mapping[str, DynamicConnectionConfig] | None = None,
+    ) -> None:
+        """Add a task to the helper's pipeline graph.
+
+        Parameters
+        ----------
+        label : `str`, optional
+            Label for the task.  If not provided, the task name will be
+            ``task_auto{self.last_auto_task_index}``, with that variable
+            incremented.
+        task_class : `type`, optional
+            Subclass of `DynamicTestPipelineTask` to use.
+        config : `DynamicTestPipelineTaskConfig`, optional
+            Task configuration to use.  Note that the dimensions are always
+            overridden by the ``dimensions`` argument and ``inputs`` and
+            ``outputs`` are updated by those arguments unless they are
+            explicitly set to empty dictionaries.
+        dimensions : `~collections.abc.Iterable` [ `str` ], optional
+            Dimensions of the task and any automatically-added input or output
+            connection.
+        inputs : `~collections.abc.Mapping` [ `str`, \
+                `DynamicConnectionConfig` ], optional
+            Input connections to add.  If not provided, a single connection is
+            added with the same dimensions as the task and dataset type name
+            ``dataset_auto{self.last_auto_dataset_type_index}``.
+        outputs : `~collections.abc.Mapping` [ `str`, \
+                `DynamicConnectionConfig` ], optional
+            Output connections to add.  If not provided, a single connection is
+            added with the same dimensions as the task and dataset type name
+            ``dataset_auto{self.last_auto_dataset_type_index}``, with that
+            variable incremented first.
+        prerequisite_inputs : `~collections.abc.Mapping` [ `str`, \
+                `DynamicConnectionConfig` ], optional
+            Prerequisite input connections to add.  Defaults to an empty
+            mapping.
+        init_inputs : `~collections.abc.Mapping` [ `str`, \
+                `DynamicConnectionConfig` ], optional
+            Init input connections to add.  Defaults to an empty mapping.
+        init_outputs : `~collections.abc.Mapping` [ `str`, \
+                `DynamicConnectionConfig` ], optional
+            Init output connections to add.  Defaults to an empty mapping.
+
+        Notes
+        -----
+        The defaults for this method's arguments are designed to allow it to be
+        called in succession to create a sequence of "one-to-one" tasks in
+        which each consumes the output of the last.
+        """
+        if config is None:
+            config = DynamicTestPipelineTaskConfig()
+        if dimensions is not None:
+            config.dimensions = list(dimensions)
+        if inputs is not None:
+            config.inputs.update(inputs)
+        else:
+            config.inputs["input_connection"] = DynamicConnectionConfig(
+                dataset_type_name=f"dataset_auto{self.last_auto_dataset_type_index}",
+                dimensions=list(config.dimensions),
+            )
+        if outputs is not None:
+            config.outputs.update(outputs)
+        else:
+            self.last_auto_dataset_type_index += 1
+            config.outputs["output_connection"] = DynamicConnectionConfig(
+                dataset_type_name=f"dataset_auto{self.last_auto_dataset_type_index}",
+                dimensions=list(config.dimensions),
+            )
+        if prerequisite_inputs is not None:
+            config.prerequisite_inputs.update(prerequisite_inputs)
+        if init_inputs is not None:
+            config.init_inputs.update(init_inputs)
+        if init_outputs is not None:
+            config.init_outputs.update(init_outputs)
+        if label is None:
+            self.last_auto_task_index += 1
+            label = f"task_auto{self.last_auto_task_index}"
+        self.pipeline_graph.add_task(label, task_class=task_class, config=config)
+
+    def make_quantum_graph_builder(
+        self,
+        *,
+        insert_mocked_inputs: bool = True,
+        register_output_dataset_types: bool = True,
+    ) -> AllDimensionsQuantumGraphBuilder:
+        """Make a quantum graph builder from the pipeline task and internal
+        data repository.
+
+        Parameters
+        ----------
+        insert_mocked_inputs : `bool`, optional
+            Whether to automatically insert datasets for all overall inputs to
+            the pipeline graph whose dataset types have not already been
+            registered.  If set to `False`, inputs must be provided by imported
+            YAML files or explicit calls to `insert_datasets`, which provides
+            more fine-grained control over the data IDs of the datasets.
+        register_output_dataset_types : `bool`, optional
+            If `True`, register all output dataset types.
+
+        Returns
+        -------
+        builder : \
+                `..all_dimensions_quantum_graph_builder.AllDimensionsQuantumGraphBuilder`
+            Quantum graph builder.  Note that
+            ``attach_datastore_records=False`` must be passed to `build`, since
+            the helper's butler does not have a datastore.
+        """
+        if insert_mocked_inputs:
+            self.pipeline_graph.resolve(self.butler.registry)
+            for _, dataset_type_node in self.pipeline_graph.iter_overall_inputs():
+                assert dataset_type_node is not None, "pipeline graph is resolved."
+                if self.butler.registry.registerDatasetType(dataset_type_node.dataset_type):
+                    self.insert_datasets(dataset_type_node.dataset_type, register=False)
+        builder = AllDimensionsQuantumGraphBuilder(
+            self.pipeline_graph,
+            self.butler,
+            input_collections=self.input_chain,
+            output_run=self.output_run,
+        )
+        if register_output_dataset_types:
+            self.pipeline_graph.register_dataset_types(self.butler)
+        return builder
+
+    def insert_datasets(
+        self, dataset_type: DatasetType | str, register: bool = True, *args: Any, **kwargs: Any
+    ) -> list[DatasetRef]:
+        """Insert input datasets into the test repository.
+
+        Parameters
+        ----------
+        dataset_type : `~lsst.daf.butler.DatasetType` or `str`
+            Dataset type or name.  If a name, it must be included in the
+            pipeline graph.
+        register : `bool`, optional
+            Whether to register the dataset type.  If `False`, the dataset type
+            must already be registered.
+        *args : `object`
+            Forwarded to `~lsst.daf.butler.query_data_ids`.
+        **kwargs : `object`
+            Forwarded to `~lsst.daf.butler.query_data_ids`.
+
+        Returns
+        -------
+        refs : `list` [ `lsst.daf.butler.DatasetRef` ]
+            References to the inserted datasets.
+
+        Notes
+        -----
+        For dataset types with dimensions that are queryable, this queries for
+        all data IDs in the repository (forwarding ``*args`` and ``**kwargs``
+        for e.g. ``where`` strings).  For skypix dimensions, this queries for
+        both patches and visit-detector regions (forwarding `*args`` and
+        ``**kwargs`` to both) and uses all overlapping sky pixels.  Dataset
+        types with a mix of skypix and queryable dimensions are not supported.
+        """
+        if isinstance(dataset_type, str):
+            self.pipeline_graph.resolve(self.butler.registry)
+            dataset_type = self.pipeline_graph.dataset_types[dataset_type].dataset_type
+        if register:
+            self.butler.registry.registerDatasetType(dataset_type)
+        dimensions = dataset_type.dimensions
+        if dataset_type.dimensions.skypix:
+            if len(dimensions) == 1:
+                (skypix_name,) = dimensions.skypix
+                pixelization = dimensions.universe.skypix_dimensions[skypix_name].pixelization
+                ranges = RangeSet()
+                for patch_record in self.butler.query_dimension_records(
+                    "patch", *args, **kwargs, explain=False
+                ):
+                    ranges |= pixelization.envelope(patch_record.region)
+                for vdr_record in self.butler.query_dimension_records(
+                    "visit_detector_region", *args, **kwargs, explain=False
+                ):
+                    ranges |= pixelization.envelope(vdr_record.region)
+                data_ids = []
+                for begin, end in ranges:
+                    for index in range(begin, end):
+                        data_ids.append(DataCoordinate.from_required_values(dimensions, (index,)))
+            else:
+                raise NotImplementedError(
+                    "Can only generate data IDs for queryable dimensions and isolated skypix."
+                )
+        else:
+            data_ids = self.butler.query_data_ids(dimensions, *args, **kwargs, explain=False)
+        return self.butler.registry.insertDatasets(dataset_type, data_ids, run=self.input_run)
+
+    def make_limited_butler(self) -> InMemoryLimitedButler:
+        """Make a test limited butler for execution.
+
+        Returns
+        -------
+        limited_butler : `.InMemoryLimitedButler`
+            A limited butler that can be used for task execution.
+
+        Notes
+        -----
+        This queries the database-only butler used for quantum-graph generation
+        for all datasets in the ``input_chain`` collection, and populates the
+        limited butler with those that have a mock storage class.  Other
+        datasets are ignored, so they will appear as though they were present
+        during quantum graph generation but absent during execution.
+        """
+        butler = InMemoryLimitedButler(self.butler.dimensions, self.butler.registry.queryDatasetTypes())
+        for ref in self.butler._query_all_datasets(self.input_chain):
+            if is_mock_name(ref.datasetType.storageClass_name):
+                butler.put(
+                    MockDataset(
+                        dataset_id=ref.id,
+                        dataset_type=ref.datasetType.to_simple(),
+                        data_id=dict(ref.dataId.mapping),
+                        run=ref.run,
+                    ),
+                    ref,
+                )
+        return butler
+
+    def make_single_quantum_executor(self) -> tuple[SingleQuantumExecutor, InMemoryLimitedButler]:
+        """Make a single-quantum executor backed by a new limited butler.
+
+        Returns
+        -------
+        executor : `..single_quantum_executor.SingleQuantumExecutor`
+            An executor for a single quantum.
+        butler : `.InMemoryLimitedButler`
+            The butler that the executor will write to.
+        """
+        butler = self.make_limited_butler()
+        return SingleQuantumExecutor(limited_butler_factory=butler.factory), butler