feat: delay dandischema.models import in dandi/files/zarr.py

Author: candleindark

dandi/files/zarr.py

@@ -11,9 +11,8 @@
 import os.path
 from pathlib import Path
 from time import sleep
-from typing import Any, Optional
+from typing import TYPE_CHECKING, Any, Optional
 
-from dandischema.models import BareAsset, DigestType
 from pydantic import BaseModel, ConfigDict, ValidationError
 import requests
 from zarr_checksum.tree import ZarrChecksumTree
@@ -55,6 +54,9 @@
     Validator,
 )
 
+if TYPE_CHECKING:
+    from dandischema.models import BareAsset
+
 lgr = get_logger()
 
 
@@ -320,6 +322,8 @@ def get_digest(self) -> Digest:
         directory, the algorithm will be the DANDI Zarr checksum algorithm; if
         it is a file, it will be MD5.
         """
+        from dandischema.models import DigestType
+
         # Avoid heavy import by importing within function:
         from dandi.support.digests import get_digest, get_zarr_checksum