From 87c0932818c470c267dda382a4cf90fe477d2728 Mon Sep 17 00:00:00 2001
From: Lorenzo Pellegrini <lrzpellegrini@gmail.com>
Date: Thu, 25 Jan 2024 12:08:52 +0100
Subject: [PATCH] Fix import issue for Python 3.8

---
 avalanche/checkpointing/checkpoint.py           | 2 +-
 avalanche/checkpointing/checkpoint_internals.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/avalanche/checkpointing/checkpoint.py b/avalanche/checkpointing/checkpoint.py
index 96d770da8..fbf6edf39 100644
--- 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
index d6aa93d02..9ccc59012 100644
--- 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