lsst-pipe-base 30.0.0rc2__py3-none-any.whl → 30.0.0rc3__py3-none-any.whl

lsst/pipe/base/quantum_graph/aggregator/_communicators.py

@@ -51,16 +51,17 @@ import time
 import uuid
 from abc import ABC, abstractmethod
 from collections.abc import Callable, Iterable, Iterator
-from contextlib import AbstractContextManager, ExitStack, contextmanager
+from contextlib import ExitStack
 from traceback import format_exception
 from types import TracebackType
 from typing import Any, Literal, Self, TypeAlias, TypeVar, Union
 
-from lsst.utils.logging import VERBOSE, LsstLogAdapter
+from lsst.utils.logging import LsstLogAdapter
 
+from .._provenance import ProvenanceQuantumScanData
 from ._config import AggregatorConfig
 from ._progress import ProgressManager, make_worker_log
-from ._structs import IngestRequest, ScanReport, WriteRequest
+from ._structs import IngestRequest, ScanReport
 
 _T = TypeVar("_T")
 
@@ -361,9 +362,9 @@ class SupervisorCommunicator:
         # scanner and the supervisor send one sentinal when done, and the
         # writer waits for (n_scanners + 1) sentinals to arrive before it
         # starts its shutdown.
-        self._write_requests: Queue[WriteRequest | Literal[_Sentinel.NO_MORE_WRITE_REQUESTS]] | None = (
-            context.make_queue() if config.output_path is not None else None
-        )
+        self._write_requests: (
+            Queue[ProvenanceQuantumScanData | Literal[_Sentinel.NO_MORE_WRITE_REQUESTS]] | None
+        ) = context.make_queue() if config.output_path is not None else None
         # All other workers use this queue to send many different kinds of
         # reports the supervisor.  The supervisor waits for a _DONE sentinal
         # from each worker before it finishes its shutdown.
@@ -461,12 +462,12 @@ class SupervisorCommunicator:
         """
         self._scan_requests.put(_ScanRequest(quantum_id), block=False)
 
-    def request_write(self, request: WriteRequest) -> None:
+    def request_write(self, request: ProvenanceQuantumScanData) -> None:
         """Send a request to the writer to write provenance for the given scan.
 
         Parameters
         ----------
-        request : `WriteRequest`
+        request : `ProvenanceQuantumScanData`
             Information from scanning a quantum (or knowing you don't have to,
             in the case of blocked quanta).
         """
@@ -621,6 +622,11 @@ class WorkerCommunicator:
         self._exit_stack.__exit__(exc_type, exc_value, traceback)
         return True
 
+    @property
+    def exit_stack(self) -> ExitStack:
+        """A `contextlib.ExitStack` tied to the communicator."""
+        return self._exit_stack
+
     def log_progress(self, level: int, message: str) -> None:
         """Send a high-level log message to the supervisor.
 
@@ -633,44 +639,6 @@ class WorkerCommunicator:
         """
         self._reports.put(_ProgressLog(message=message, level=level), block=False)
 
-    def enter(
-        self,
-        cm: AbstractContextManager[_T],
-        on_close: str | None = None,
-        level: int = VERBOSE,
-        is_progress_log: bool = False,
-    ) -> _T:
-        """Enter a context manager that will be exited when the communicator's
-        context is exited.
-
-        Parameters
-        ----------
-        cm : `contextlib.AbstractContextManager`
-            A context manager to enter.
-        on_close : `str`, optional
-            A log message to emit (on the worker's logger) just before the
-            given context manager is exited.  This can be used to indicate
-            what's going on when an ``__exit__`` implementation has a lot of
-            work to do (e.g. moving a large file into a zip archive).
-        level : `int`, optional
-            Level for the ``on_close`` log message.
-        is_progress_log : `bool`, optional
-            If `True`, send the ``on_close`` message to the supervisor via
-            `log_progress` as well as the worker's logger.
-        """
-        if on_close is None:
-            return self._exit_stack.enter_context(cm)
-
-        @contextmanager
-        def wrapper() -> Iterator[_T]:
-            with cm as result:
-                yield result
-                self.log.log(level, on_close)
-                if is_progress_log:
-                    self.log_progress(level, on_close)
-
-        return self._exit_stack.enter_context(wrapper())
-
     def check_for_cancel(self) -> None:
         """Check for a cancel signal from the supervisor and raise
         `FatalWorkerError` if it is present.
@@ -728,12 +696,12 @@ class ScannerCommunicator(WorkerCommunicator):
         else:
             self._reports.put(_IngestReport(1), block=False)
 
-    def request_write(self, request: WriteRequest) -> None:
+    def request_write(self, request: ProvenanceQuantumScanData) -> None:
         """Ask the writer to write provenance for a quantum.
 
         Parameters
         ----------
-        request : `WriteRequest`
+        request : `ProvenanceQuantumScanData`
             Result of scanning a quantum.
         """
         assert self._write_requests is not None, "Writer should not be used if writing is disabled."
@@ -913,12 +881,12 @@ class WriterCommunicator(WorkerCommunicator):
         self._reports.put(_Sentinel.WRITER_DONE, block=False)
         return result
 
-    def poll(self) -> Iterator[WriteRequest]:
+    def poll(self) -> Iterator[ProvenanceQuantumScanData]:
         """Poll for writer requests from the scanner workers and supervisor.
 
         Yields
         ------
-        request : `WriteRequest`
+        request : `ProvenanceQuantumScanData`
             The result of a quantum scan.
 
         Notes

lsst/pipe/base/quantum_graph/aggregator/_scanner.py

@@ -38,23 +38,19 @@ from typing import Any, Literal, Self
 import zstandard
 
 from lsst.daf.butler import ButlerLogRecords, DatasetRef, QuantumBackedButler
-from lsst.utils.iteration import ensure_iterable
 
 from ... import automatic_connection_constants as acc
-from ..._status import ExceptionInfo, QuantumAttemptStatus, QuantumSuccessCaveats
 from ..._task_metadata import TaskMetadata
-from ...log_capture import _ExecutionLogRecordsExtra
 from ...pipeline_graph import PipelineGraph, TaskImportMode
-from ...resource_usage import QuantumResourceUsage
 from .._multiblock import Compressor
 from .._predicted import (
     PredictedDatasetModel,
     PredictedQuantumDatasetsModel,
     PredictedQuantumGraphReader,
 )
-from .._provenance import ProvenanceInitQuantumModel, ProvenanceQuantumAttemptModel, ProvenanceQuantumModel
+from .._provenance import ProvenanceQuantumScanModels, ProvenanceQuantumScanStatus
 from ._communicators import ScannerCommunicator
-from ._structs import IngestRequest, InProgressScan, ScanReport, ScanStatus, WriteRequest
+from ._structs import IngestRequest, ScanReport
 
 
 @dataclasses.dataclass
@@ -94,7 +90,7 @@ class Scanner(AbstractContextManager):
         if self.comms.config.mock_storage_classes:
             import lsst.pipe.base.tests.mocks  # noqa: F401
         self.comms.log.verbose("Reading from predicted quantum graph.")
-        self.reader = self.comms.enter(
+        self.reader = self.comms.exit_stack.enter_context(
             PredictedQuantumGraphReader.open(self.predicted_path, import_mode=TaskImportMode.DO_NOT_IMPORT)
         )
         self.reader.read_dimension_data()
@@ -196,7 +192,7 @@ class Scanner(AbstractContextManager):
         ref = self.reader.components.make_dataset_ref(predicted)
         return self.qbb.stored(ref)
 
-    def scan_quantum(self, quantum_id: uuid.UUID) -> InProgressScan:
+    def scan_quantum(self, quantum_id: uuid.UUID) -> ProvenanceQuantumScanModels:
         """Scan for a quantum's completion and error status, and its output
         datasets' existence.
 
@@ -207,76 +203,38 @@ class Scanner(AbstractContextManager):
 
         Returns
         -------
-        result : `InProgressScan`
+        result : `ProvenanceQuantumScanModels`
             Scan result struct.
         """
         if (predicted_quantum := self.init_quanta.get(quantum_id)) is not None:
-            result = InProgressScan(predicted_quantum.quantum_id, status=ScanStatus.INIT)
+            result = ProvenanceQuantumScanModels(
+                predicted_quantum.quantum_id, status=ProvenanceQuantumScanStatus.INIT
+            )
             self.comms.log.debug("Created init scan for %s (%s)", quantum_id, predicted_quantum.task_label)
         else:
             self.reader.read_quantum_datasets([quantum_id])
-            predicted_quantum = self.reader.components.quantum_datasets[quantum_id]
+            predicted_quantum = self.reader.components.quantum_datasets.pop(quantum_id)
             self.comms.log.debug(
                 "Scanning %s (%s@%s)",
                 quantum_id,
                 predicted_quantum.task_label,
                 predicted_quantum.data_coordinate,
             )
-            result = InProgressScan(predicted_quantum.quantum_id, ScanStatus.INCOMPLETE)
-            del self.reader.components.quantum_datasets[quantum_id]
-            last_attempt = ProvenanceQuantumAttemptModel()
-            if not self._read_log(predicted_quantum, result, last_attempt):
-                self.comms.log.debug("Abandoning scan for %s; no log dataset.", quantum_id)
-                self.comms.report_scan(ScanReport(result.quantum_id, result.status))
-                return result
-            if not self._read_metadata(predicted_quantum, result, last_attempt):
-                # We found the log dataset, but no metadata; this means the
-                # quantum failed, but a retry might still happen that could
-                # turn it into a success if we can't yet assume the run is
-                # complete.
-                self.comms.log.debug("Abandoning scan for %s.", quantum_id)
+            logs = self._read_log(predicted_quantum)
+            metadata = self._read_metadata(predicted_quantum)
+            result = ProvenanceQuantumScanModels.from_metadata_and_logs(
+                predicted_quantum, metadata, logs, assume_complete=self.comms.config.assume_complete
+            )
+            if result.status is ProvenanceQuantumScanStatus.ABANDONED:
+                self.comms.log.debug("Abandoning scan for failed quantum %s.", quantum_id)
                 self.comms.report_scan(ScanReport(result.quantum_id, result.status))
                 return result
-            last_attempt.attempt = len(result.attempts)
-            result.attempts.append(last_attempt)
-        assert result.status is not ScanStatus.INCOMPLETE
-        assert result.status is not ScanStatus.ABANDONED
-
-        if len(result.logs.attempts) < len(result.attempts):
-            # Logs were not found for this attempt; must have been a hard error
-            # that kept the `finally` block from running or otherwise
-            # interrupted the writing of the logs.
-            result.logs.attempts.append(None)
-            if result.status is ScanStatus.SUCCESSFUL:
-                # But we found the metadata!  Either that hard error happened
-                # at a very unlucky time (in between those two writes), or
-                # something even weirder happened.
-                result.attempts[-1].status = QuantumAttemptStatus.LOGS_MISSING
-            else:
-                result.attempts[-1].status = QuantumAttemptStatus.FAILED
-        if len(result.metadata.attempts) < len(result.attempts):
-            # Metadata missing usually just means a failure.  In any case, the
-            # status will already be correct, either because it was set to a
-            # failure when we read the logs, or left at UNKNOWN if there were
-            # no logs.  Note that scanners never process BLOCKED quanta at all.
-            result.metadata.attempts.append(None)
-        assert len(result.logs.attempts) == len(result.attempts) or len(result.metadata.attempts) == len(
-            result.attempts
-        ), (
-            "The only way we can add more than one quantum attempt is by "
-            "extracting info stored with the logs, and that always appends "
-            "a log attempt and a metadata attempt, so this must be a bug in "
-            "the scanner."
-        )
-        # Scan for output dataset existence, skipping any the metadata reported
-        # on as well as and the metadata and logs themselves (since we just
-        # checked those).
         for predicted_output in itertools.chain.from_iterable(predicted_quantum.outputs.values()):
-            if predicted_output.dataset_id not in result.outputs:
-                result.outputs[predicted_output.dataset_id] = self.scan_dataset(predicted_output)
+            if predicted_output.dataset_id not in result.output_existence:
+                result.output_existence[predicted_output.dataset_id] = self.scan_dataset(predicted_output)
         to_ingest = self._make_ingest_request(predicted_quantum, result)
         if self.comms.config.output_path is not None:
-            to_write = self._make_write_request(predicted_quantum, result)
+            to_write = result.to_scan_data(predicted_quantum, compressor=self.compressor)
             self.comms.request_write(to_write)
         self.comms.request_ingest(to_ingest)
         self.comms.report_scan(ScanReport(result.quantum_id, result.status))
@@ -284,7 +242,7 @@ class Scanner(AbstractContextManager):
         return result
 
     def _make_ingest_request(
-        self, predicted_quantum: PredictedQuantumDatasetsModel, result: InProgressScan
+        self, predicted_quantum: PredictedQuantumDatasetsModel, result: ProvenanceQuantumScanModels
     ) -> IngestRequest:
         """Make an ingest request from a quantum scan.
 
@@ -292,7 +250,7 @@ class Scanner(AbstractContextManager):
         ----------
         predicted_quantum : `PredictedQuantumDatasetsModel`
             Information about the predicted quantum.
-        result : `InProgressScan`
+        result : `ProvenanceQuantumScanModels`
             Result of a quantum scan.
 
         Returns
@@ -305,7 +263,7 @@ class Scanner(AbstractContextManager):
         }
         to_ingest_predicted: list[PredictedDatasetModel] = []
         to_ingest_refs: list[DatasetRef] = []
-        for dataset_id, was_produced in result.outputs.items():
+        for dataset_id, was_produced in result.output_existence.items():
             if was_produced:
                 predicted_output = predicted_outputs_by_id[dataset_id]
                 to_ingest_predicted.append(predicted_output)
@@ -313,69 +271,18 @@ class Scanner(AbstractContextManager):
         to_ingest_records = self.qbb._datastore.export_predicted_records(to_ingest_refs)
         return IngestRequest(result.quantum_id, to_ingest_predicted, to_ingest_records)
 
-    def _make_write_request(
-        self, predicted_quantum: PredictedQuantumDatasetsModel, result: InProgressScan
-    ) -> WriteRequest:
-        """Make a write request from a quantum scan.
+    def _read_metadata(self, predicted_quantum: PredictedQuantumDatasetsModel) -> TaskMetadata | None:
+        """Attempt to read the metadata dataset for a quantum.
 
         Parameters
         ----------
         predicted_quantum : `PredictedQuantumDatasetsModel`
             Information about the predicted quantum.
-        result : `InProgressScan`
-            Result of a quantum scan.
 
         Returns
         -------
-        write_request : `WriteRequest`
-            A request to be sent to the writer.
-        """
-        quantum: ProvenanceInitQuantumModel | ProvenanceQuantumModel
-        if result.status is ScanStatus.INIT:
-            quantum = ProvenanceInitQuantumModel.from_predicted(predicted_quantum)
-        else:
-            quantum = ProvenanceQuantumModel.from_predicted(predicted_quantum)
-            quantum.attempts = result.attempts
-        request = WriteRequest(
-            result.quantum_id,
-            result.status,
-            existing_outputs={
-                dataset_id for dataset_id, was_produced in result.outputs.items() if was_produced
-            },
-            quantum=quantum.model_dump_json().encode(),
-            logs=result.logs.model_dump_json().encode() if result.logs.attempts else b"",
-            metadata=result.metadata.model_dump_json().encode() if result.metadata.attempts else b"",
-        )
-        if self.compressor is not None:
-            request.quantum = self.compressor.compress(request.quantum)
-            request.logs = self.compressor.compress(request.logs) if request.logs else b""
-            request.metadata = self.compressor.compress(request.metadata) if request.metadata else b""
-            request.is_compressed = True
-        return request
-
-    def _read_metadata(
-        self,
-        predicted_quantum: PredictedQuantumDatasetsModel,
-        result: InProgressScan,
-        last_attempt: ProvenanceQuantumAttemptModel,
-    ) -> bool:
-        """Attempt to read the metadata dataset for a quantum to extract
-        provenance information from it.
-
-        Parameters
-        ----------
-        predicted_quantum : `PredictedQuantumDatasetsModel`
-            Information about the predicted quantum.
-        result : `InProgressScan`
-            Result object to be modified in-place.
-        last_attempt : `ScanningProvenanceQuantumAttemptModel`
-            Structure to fill in with information about the last attempt to
-            run this quantum.
-
-        Returns
-        -------
-        complete : `bool`
-            Whether the quantum is complete.
+        metadata : `...TaskMetadata` or `None`
+            Task metadata.
         """
         (predicted_dataset,) = predicted_quantum.outputs[acc.METADATA_OUTPUT_CONNECTION_NAME]
         ref = self.reader.components.make_dataset_ref(predicted_dataset)
@@ -383,129 +290,28 @@ class Scanner(AbstractContextManager):
             # This assumes QBB metadata writes are atomic, which should be the
             # case. If it's not we'll probably get pydantic validation errors
             # here.
-            metadata: TaskMetadata = self.qbb.get(ref, storageClass="TaskMetadata")
+            return self.qbb.get(ref, storageClass="TaskMetadata")
         except FileNotFoundError:
-            result.outputs[ref.id] = False
-            if self.comms.config.assume_complete:
-                result.status = ScanStatus.FAILED
-            else:
-                result.status = ScanStatus.ABANDONED
-                return False
-        else:
-            result.status = ScanStatus.SUCCESSFUL
-            result.outputs[ref.id] = True
-            last_attempt.status = QuantumAttemptStatus.SUCCESSFUL
-            try:
-                # Int conversion guards against spurious conversion to
-                # float that can apparently sometimes happen in
-                # TaskMetadata.
-                last_attempt.caveats = QuantumSuccessCaveats(int(metadata["quantum"]["caveats"]))
-            except LookupError:
-                pass
-            try:
-                last_attempt.exception = ExceptionInfo._from_metadata(
-                    metadata[predicted_quantum.task_label]["failure"]
-                )
-            except LookupError:
-                pass
-            try:
-                for id_str in ensure_iterable(metadata["quantum"].getArray("outputs")):
-                    result.outputs[uuid.UUID(id_str)]
-            except LookupError:
-                pass
-            else:
-                # If the metadata told us what it wrote, anything not in that
-                # list was not written.
-                for predicted_output in itertools.chain.from_iterable(predicted_quantum.outputs.values()):
-                    result.outputs.setdefault(predicted_output.dataset_id, False)
-            last_attempt.resource_usage = QuantumResourceUsage.from_task_metadata(metadata)
-            result.metadata.attempts.append(metadata)
-        return True
-
-    def _read_log(
-        self,
-        predicted_quantum: PredictedQuantumDatasetsModel,
-        result: InProgressScan,
-        last_attempt: ProvenanceQuantumAttemptModel,
-    ) -> bool:
-        """Attempt to read the log dataset for a quantum to test for the
-        quantum's completion (the log is always written last) and aggregate
-        the log content in the provenance quantum graph.
+            return None
+
+    def _read_log(self, predicted_quantum: PredictedQuantumDatasetsModel) -> ButlerLogRecords | None:
+        """Attempt to read the log dataset for a quantum.
 
         Parameters
         ----------
         predicted_quantum : `PredictedQuantumDatasetsModel`
             Information about the predicted quantum.
-        result : `InProgressScan`
-            Result object to be modified in-place.
-        last_attempt : `ScanningProvenanceQuantumAttemptModel`
-            Structure to fill in with information about the last attempt to
-            run this quantum.
 
         Returns
         -------
-        complete : `bool`
-            Whether the quantum is complete.
+        logs : `lsst.daf.butler.logging.ButlerLogRecords` or `None`
+            Task logs.
         """
         (predicted_dataset,) = predicted_quantum.outputs[acc.LOG_OUTPUT_CONNECTION_NAME]
         ref = self.reader.components.make_dataset_ref(predicted_dataset)
         try:
             # This assumes QBB log writes are atomic, which should be the case.
             # If it's not we'll probably get pydantic validation errors here.
-            log_records: ButlerLogRecords = self.qbb.get(ref)
+            return self.qbb.get(ref)
         except FileNotFoundError:
-            result.outputs[ref.id] = False
-            if self.comms.config.assume_complete:
-                result.status = ScanStatus.FAILED
-            else:
-                result.status = ScanStatus.ABANDONED
-                return False
-        else:
-            # Set the attempt's run status to FAILED, since the default is
-            # UNKNOWN (i.e. logs *and* metadata are missing) and we now know
-            # the logs exist.  This will usually get replaced by SUCCESSFUL
-            # when we look for metadata next.
-            last_attempt.status = QuantumAttemptStatus.FAILED
-            result.outputs[ref.id] = True
-            if log_records.extra:
-                log_extra = _ExecutionLogRecordsExtra.model_validate(log_records.extra)
-                self._extract_from_log_extra(log_extra, result, last_attempt=last_attempt)
-            result.logs.attempts.append(list(log_records))
-        return True
-
-    def _extract_from_log_extra(
-        self,
-        log_extra: _ExecutionLogRecordsExtra,
-        result: InProgressScan,
-        last_attempt: ProvenanceQuantumAttemptModel | None,
-    ) -> None:
-        for previous_attempt_log_extra in log_extra.previous_attempts:
-            self._extract_from_log_extra(previous_attempt_log_extra, result, last_attempt=None)
-        quantum_attempt: ProvenanceQuantumAttemptModel
-        if last_attempt is None:
-            # This is not the last attempt, so it must be a failure.
-            quantum_attempt = ProvenanceQuantumAttemptModel(
-                attempt=len(result.attempts), status=QuantumAttemptStatus.FAILED
-            )
-            # We also need to get the logs from this extra provenance, since
-            # they won't be the main section of the log records.
-            result.logs.attempts.append(log_extra.logs)
-            # The special last attempt is only appended after we attempt to
-            # read metadata later, but we have to append this one now.
-            result.attempts.append(quantum_attempt)
-        else:
-            assert not log_extra.logs, "Logs for the last attempt should not be stored in the extra JSON."
-            quantum_attempt = last_attempt
-        if log_extra.exception is not None or log_extra.metadata is not None or last_attempt is None:
-            # We won't be getting a separate metadata dataset, so anything we
-            # might get from the metadata has to come from this extra
-            # provenance in the logs.
-            quantum_attempt.exception = log_extra.exception
-            if log_extra.metadata is not None:
-                quantum_attempt.resource_usage = QuantumResourceUsage.from_task_metadata(log_extra.metadata)
-                result.metadata.attempts.append(log_extra.metadata)
-            else:
-                result.metadata.attempts.append(None)
-        # Regardless of whether this is the last attempt or not, we can only
-        # get the previous_process_quanta from the log extra.
-        quantum_attempt.previous_process_quanta.extend(log_extra.previous_process_quanta)
+            return None

lsst/pipe/base/quantum_graph/aggregator/_structs.py

@@ -27,68 +27,16 @@
 
 from __future__ import annotations
 
-__all__ = (
-    "InProgressScan",
-    "IngestRequest",
-    "ScanReport",
-    "ScanStatus",
-    "WriteRequest",
-)
+__all__ = ("IngestRequest", "ScanReport")
 
 import dataclasses
-import enum
 import uuid
 
 from lsst.daf.butler.datastore.record_data import DatastoreRecordData
 
 from .._common import DatastoreName
 from .._predicted import PredictedDatasetModel
-from .._provenance import (
-    ProvenanceLogRecordsModel,
-    ProvenanceQuantumAttemptModel,
-    ProvenanceTaskMetadataModel,
-)
-
-
-class ScanStatus(enum.Enum):
-    """Status enum for quantum scanning.
-
-    Note that this records the status for the *scanning* which is distinct
-    from the status of the quantum's execution.
-    """
-
-    INCOMPLETE = enum.auto()
-    """The quantum is not necessarily done running, and cannot be scanned
-    conclusively yet.
-    """
-
-    ABANDONED = enum.auto()
-    """The quantum's execution appears to have failed but we cannot rule out
-    the possibility that it could be recovered, but we've also waited long
-    enough (according to `ScannerTimeConfigDict.retry_timeout`) that it's time
-    to stop trying for now.
-
-    This state means a later run with `ScannerConfig.assume_complete` is
-    required.
-    """
-
-    SUCCESSFUL = enum.auto()
-    """The quantum was conclusively scanned and was executed successfully,
-    unblocking scans for downstream quanta.
-    """
-
-    FAILED = enum.auto()
-    """The quantum was conclusively scanned and failed execution, blocking
-    scans for downstream quanta.
-    """
-
-    BLOCKED = enum.auto()
-    """A quantum upstream of this one failed."""
-
-    INIT = enum.auto()
-    """Init quanta need special handling, because they don't have logs and
-    metadata.
-    """
+from .._provenance import ProvenanceQuantumScanStatus
 
 
 @dataclasses.dataclass
@@ -98,7 +46,7 @@ class ScanReport:
     quantum_id: uuid.UUID
     """Unique ID of the quantum."""
 
-    status: ScanStatus
+    status: ProvenanceQuantumScanStatus
     """Combined status of the scan and the execution of the quantum."""
 
 
@@ -117,61 +65,3 @@ class IngestRequest:
 
     def __bool__(self) -> bool:
         return bool(self.datasets or self.records)
-
-
-@dataclasses.dataclass
-class InProgressScan:
-    """A struct that represents a quantum that is being scanned."""
-
-    quantum_id: uuid.UUID
-    """Unique ID for the quantum."""
-
-    status: ScanStatus
-    """Combined status for the scan and the execution of the quantum."""
-
-    attempts: list[ProvenanceQuantumAttemptModel] = dataclasses.field(default_factory=list)
-    """Provenance information about each attempt to run the quantum."""
-
-    outputs: dict[uuid.UUID, bool] = dataclasses.field(default_factory=dict)
-    """Unique IDs of the output datasets mapped to whether they were actually
-    produced.
-    """
-
-    metadata: ProvenanceTaskMetadataModel = dataclasses.field(default_factory=ProvenanceTaskMetadataModel)
-    """Task metadata information for each attempt.
-    """
-
-    logs: ProvenanceLogRecordsModel = dataclasses.field(default_factory=ProvenanceLogRecordsModel)
-    """Log records for each attempt.
-    """
-
-
-@dataclasses.dataclass
-class WriteRequest:
-    """A struct that represents a request to write provenance for a quantum."""
-
-    quantum_id: uuid.UUID
-    """Unique ID for the quantum."""
-
-    status: ScanStatus
-    """Combined status for the scan and the execution of the quantum."""
-
-    existing_outputs: set[uuid.UUID] = dataclasses.field(default_factory=set)
-    """Unique IDs of the output datasets that were actually written."""
-
-    quantum: bytes = b""
-    """Serialized quantum provenance model.
-
-    This may be empty for quanta that had no attempts.
-    """
-
-    metadata: bytes = b""
-    """Serialized task metadata."""
-
-    logs: bytes = b""
-    """Serialized logs."""
-
-    is_compressed: bool = False
-    """Whether the `quantum`, `metadata`, and `log` attributes are
-    compressed.
-    """