Fix: Literal import for python < 3.9.

Labbeti · Apr 17, 2024 · 0000bcd · 0000bcd
1 parent f887423
commit 0000bcd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/aac_datasets/utils/download.py b/src/aac_datasets/utils/download.py
@@ -5,10 +5,11 @@
 import os
 import os.path as osp
 from pathlib import Path
-from typing import List, Literal, Union
+from typing import List, Union
 
 import tqdm
 from torch.hub import download_url_to_file
+from typing_extensions import Literal
 
 HASH_TYPES = ("sha256", "md5")
 DEFAULT_CHUNK_SIZE = 256 * 1024**2  # 256 MiB