Replace queryCollections call in export

lsst · Aug 15, 2024 · 67dc2f6 · 67dc2f6
1 parent 04b15dc
commit 67dc2f6
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/python/lsst/daf/butler/transfers/_context.py b/python/lsst/daf/butler/transfers/_context.py
@@ -358,11 +358,10 @@ def _computeDatasetAssociations(self) -> dict[str, list[DatasetAssociation]]:
             collectionTypes = {CollectionType.TAGGED}
             if datasetType.isCalibration():
                 collectionTypes.add(CollectionType.CALIBRATION)
-            resolved_collections = self._butler._registry.queryCollections(
+            resolved_collections = self._butler.collections.query(
                 self._collections.keys(),
-                datasetType=datasetType,
-                collectionTypes=collectionTypes,
-                flattenChains=False,
+                collection_types=collectionTypes,
+                flatten_chains=False,
             )
             with self._butler._query() as query:
                 query = query.join_dataset_search(datasetType, resolved_collections)