Use newer ref.replace API rather than overrideStorageClass

lsst · Nov 5, 2024 · 5f4f16d · 5f4f16d
1 parent 5221e53
commit 5f4f16d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/lsst/pipe/base/graph/graph.py b/python/lsst/pipe/base/graph/graph.py
@@ -1790,7 +1790,7 @@ def get_refs(
             def _update_ref(ref: DatasetRef) -> DatasetRef:
                 internal_dataset_type = dataset_types.get(ref.datasetType.name, ref.datasetType)
                 if internal_dataset_type.storageClass_name != ref.datasetType.storageClass_name:
-                    ref = ref.overrideStorageClass(internal_dataset_type.storageClass_name)
+                    ref = ref.replace(storage_class=internal_dataset_type.storageClass_name)
                 return ref
 
             # Convert output_refs to the data repository storage classes, too.