Fix typing error under mypy 0.990

dandi · Nov 11, 2022 · 3452a27 · 3452a27
1 parent c66fd18
commit 3452a27
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/dandi/files/_private.py b/dandi/files/_private.py
@@ -1,8 +1,10 @@
 from __future__ import annotations
 
+from collections.abc import Mapping
 from dataclasses import dataclass
 from enum import Enum
 from pathlib import Path
+from typing import ClassVar
 import weakref
 
 from dandi.consts import (
@@ -14,6 +16,7 @@
 
 from .bases import DandiFile, GenericAsset, LocalAsset, NWBAsset, VideoAsset
 from .bids import (
+    BIDSAsset,
     BIDSDatasetDescriptionAsset,
     GenericBIDSAsset,
     NWBBIDSAsset,
@@ -54,7 +57,7 @@ def classify(path: Path) -> DandiFileType:
 class DandiFileFactory:
     """:meta private:"""
 
-    CLASSES: dict[DandiFileType, type[LocalAsset]] = {
+    CLASSES: Mapping[DandiFileType, type[LocalAsset]] = {
         DandiFileType.NWB: NWBAsset,
         DandiFileType.ZARR: ZarrAsset,
         DandiFileType.VIDEO: VideoAsset,
@@ -74,7 +77,7 @@ class BIDSFileFactory(DandiFileFactory):
 
     bids_dataset_description: BIDSDatasetDescriptionAsset
 
-    CLASSES = {
+    CLASSES: ClassVar[Mapping[DandiFileType, type[BIDSAsset]]] = {
         DandiFileType.NWB: NWBBIDSAsset,
         DandiFileType.ZARR: ZarrBIDSAsset,
         DandiFileType.VIDEO: GenericBIDSAsset,