importing TypedDict from typing_extensions for python 3.7

neuromorphs · Dec 5, 2023 · d2c995e · d2c995e
1 parent ccafa5a
commit d2c995e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tonic/cached_dataset.py b/tonic/cached_dataset.py
@@ -1,10 +1,17 @@
 import logging
 import os
+import sys
+
+if sys.version_info >= (3, 8):
+    from typing import Callable, Iterable, Optional, Tuple, TypedDict, Union
+else:
+    from typing import Callable, Iterable, Optional, Tuple, Union
+    from typing_extensions import TypedDict
+
 import random
 import shutil
 from dataclasses import dataclass, field
 from pathlib import Path
-from typing import Callable, Iterable, Optional, Tuple, TypedDict, Union
 from warnings import warn
 
 import h5py