mirror of
https://github.com/modelscope/modelscope.git
synced 2026-02-24 04:01:10 +01:00
lint code (#1619)
This commit is contained in:
@@ -9,7 +9,6 @@ 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,
|
||||
@@ -18,7 +17,6 @@ from modelscope.hub.constants import (DEFAULT_MODELSCOPE_DOMAIN,
|
||||
MODEL_ID_SEPARATOR, MODELSCOPE_DOMAIN,
|
||||
MODELSCOPE_SDK_DEBUG,
|
||||
MODELSCOPE_URL_SCHEME)
|
||||
from modelscope.hub.errors import FileIntegrityError
|
||||
from modelscope.utils.logger import get_logger
|
||||
|
||||
logger = get_logger()
|
||||
@@ -312,6 +310,7 @@ def convert_timestamp(time_stamp: Union[int, str, datetime],
|
||||
Returns:
|
||||
Timezone-aware datetime object or None if input is None
|
||||
"""
|
||||
import zoneinfo
|
||||
if not time_stamp:
|
||||
return None
|
||||
|
||||
|
||||
@@ -5,8 +5,6 @@ import datetime
|
||||
import unittest
|
||||
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,
|
||||
@@ -236,6 +234,7 @@ class HubRepoInfoTest(unittest.TestCase):
|
||||
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
|
||||
def test_model_info_class_comprehensive(self):
|
||||
"""Test ModelInfo class initialization and properties."""
|
||||
import zoneinfo
|
||||
model_data = {
|
||||
'Id': 123,
|
||||
'Name': 'demo-model',
|
||||
@@ -344,6 +343,7 @@ class HubRepoInfoTest(unittest.TestCase):
|
||||
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
|
||||
def test_dataset_info_class(self):
|
||||
# Test DatasetInfo class initialization and properties
|
||||
import zoneinfo
|
||||
dataset_data = {
|
||||
'Id': 456,
|
||||
'Name': 'demo-dataset',
|
||||
|
||||
Reference in New Issue
Block a user