diff --git a/src/aac_datasets/utils/download.py b/src/aac_datasets/utils/download.py index 536bab8..32db976 100644 --- 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