Fix minor review comments

lsst · Sep 11, 2024 · af34091 · af34091
1 parent 701bd48
commit af34091
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/python/lsst/daf/butler/_dataset_type.py b/python/lsst/daf/butler/_dataset_type.py
@@ -805,8 +805,12 @@ def get_dataset_type_name(datasetTypeOrName: DatasetType | str) -> str:
     Parameters
     ----------
     datasetTypeOrName : `DatasetType` | `str`
-        A DatasetType, or the name of a DatasetType. This union is a common
-        parameter in many `Butler` methods.
+        A DatasetType, or the name of a DatasetType.
+
+    Returns
+    -------
+    name
+        The name associated with the given DatasetType, or the given string.
     """
     if isinstance(datasetTypeOrName, DatasetType):
         return datasetTypeOrName.name

diff --git a/python/lsst/daf/butler/registry/datasets/byDimensions/_manager.py b/python/lsst/daf/butler/registry/datasets/byDimensions/_manager.py
@@ -11,8 +11,8 @@
 
 import sqlalchemy
 
-from ...._dataset_ref import DatasetId, DatasetIdGenEnum, DatasetRef, DatasetType
-from ...._dataset_type import get_dataset_type_name
+from ...._dataset_ref import DatasetId, DatasetIdGenEnum, DatasetRef
+from ...._dataset_type import DatasetType, get_dataset_type_name
 from ...._exceptions_legacy import DatasetTypeError
 from ....dimensions import DimensionUniverse
 from ..._collection_summary import CollectionSummary