diff --git a/python/lsst/daf/butler_migrate/script/rewrite_sqlite_registry.py b/python/lsst/daf/butler_migrate/script/rewrite_sqlite_registry.py index 5413578..6488fc8 100644 --- a/python/lsst/daf/butler_migrate/script/rewrite_sqlite_registry.py +++ b/python/lsst/daf/butler_migrate/script/rewrite_sqlite_registry.py @@ -27,10 +27,17 @@ import tempfile from collections import defaultdict -from lsst.daf.butler import Butler, Config, DatasetAssociation, DatasetId, DatasetRef, SkyPixDimension +from lsst.daf.butler import ( + Butler, + CollectionType, + Config, + DatasetAssociation, + DatasetId, + DatasetRef, + SkyPixDimension, +) from lsst.daf.butler.datastores.fileDatastore import FileDatastore from lsst.daf.butler.direct_butler import DirectButler -from lsst.daf.butler.registry import CollectionType from lsst.daf.butler.registry.databases.sqlite import SqliteDatabase from lsst.daf.butler.registry.sql_registry import SqlRegistry from lsst.daf.butler.transfers import RepoExportContext