Fix import issue for Python 3.8

ContinualAI · Jan 25, 2024 · 87c0932 · 87c0932
1 parent 6836c26
commit 87c0932
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/avalanche/checkpointing/checkpoint.py b/avalanche/checkpointing/checkpoint.py
@@ -8,13 +8,13 @@
     Optional,
     Tuple,
     Type,
-    TypeAlias,
     TypeVar,
     BinaryIO,
     IO,
     Union,
     Collection,
 )
+from typing_extensions import TypeAlias
 
 import dill
 import torch

diff --git a/avalanche/checkpointing/checkpoint_internals.py b/avalanche/checkpointing/checkpoint_internals.py
@@ -7,12 +7,12 @@
     Sequence,
     Tuple,
     Type,
-    TypeAlias,
     TypeVar,
     BinaryIO,
     IO,
     Union,
 )
+from typing_extensions import TypeAlias
 
 import dill
 import torch