diff --git a/modelscope/hub/utils/utils.py b/modelscope/hub/utils/utils.py index 161f866b..4ef9e494 100644 --- a/modelscope/hub/utils/utils.py +++ b/modelscope/hub/utils/utils.py @@ -5,11 +5,11 @@ import hashlib import os import sys import time -import zoneinfo from datetime import datetime from pathlib import Path from typing import Generator, List, Optional, Union +import zoneinfo from filelock import BaseFileLock, FileLock, SoftFileLock, Timeout from modelscope.hub.constants import (DEFAULT_MODELSCOPE_DOMAIN, diff --git a/tests/hub/test_hub_repo_info.py b/tests/hub/test_hub_repo_info.py index 6fe2408f..497c1cd8 100644 --- a/tests/hub/test_hub_repo_info.py +++ b/tests/hub/test_hub_repo_info.py @@ -3,9 +3,10 @@ import datetime import unittest -import zoneinfo from unittest.mock import Mock, patch +import zoneinfo + from modelscope.hub.api import HubApi from modelscope.hub.info import BlobLfsInfo, DatasetInfo, ModelInfo from modelscope.utils.constant import (DEFAULT_DATASET_REVISION,