Add the provenance parameter to Datastore put

lsst · Jan 31, 2025 · 3c64e81 · 3c64e81
1 parent 1879d7d
commit 3c64e81
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/lsst/analysis/tools/interfaces/datastore/_sasquatchDatastore.py b/python/lsst/analysis/tools/interfaces/datastore/_sasquatchDatastore.py
@@ -39,7 +39,7 @@
 from . import SasquatchDispatcher
 
 if TYPE_CHECKING:
-    from lsst.daf.butler import Config, DatasetType, LookupKey
+    from lsst.daf.butler import Config, DatasetProvenance, DatasetType, LookupKey
     from lsst.daf.butler.registry.interfaces import DatasetIdRef, DatastoreRegistryBridgeManager
 
 
@@ -151,7 +151,9 @@ def clone(self, bridgeManager: DatastoreRegistryBridgeManager) -> SasquatchDatas
     def bridge(self) -> DatastoreRegistryBridge:
         return self._bridge
 
-    def put(self, inMemoryDataset: Any, ref: DatasetRef) -> None:
+    def put(
+        self, inMemoryDataset: Any, ref: DatasetRef, *, provenance: DatasetProvenance | None = None
+    ) -> None:
         if self.constraints.isAcceptable(ref):
             self._dispatcher.dispatchRef(inMemoryDataset, ref, extraFields=self.extra_fields)
         else: