Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

compatible with repeat download with cache

Browse Source
This commit is contained in:
mulin.lyh
2024-02-04 22:45:10 +08:00
parent 5e8a8f4e93
commit 82d598be84
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/hub/test_hub_revision.py
View File

@@ -1,6 +1,7 @@
# Copyright (c) Alibaba, Inc. and its affiliates.
import os
import tempfile
import time
import unittest
import uuid
from datetime import datetime
@@ -87,6 +88,7 @@ class HubRevisionTest(unittest.TestCase):
os.path.join(snapshot_path, download_model_file_name))
self.add_new_file_and_tag()
with tempfile.TemporaryDirectory() as temp_cache_dir:
time.sleep(11)
snapshot_path = snapshot_download(
self.model_id,
revision=self.revision,
@@ -96,6 +98,7 @@ class HubRevisionTest(unittest.TestCase):
assert not os.path.exists(
os.path.join(snapshot_path, download_model_file_name2))
with tempfile.TemporaryDirectory() as temp_cache_dir:
time.sleep(11)
snapshot_path = snapshot_download(
self.model_id,
revision=self.revision2,

7
tests/hub/test_hub_revision_release_mode.py
View File

@@ -77,6 +77,7 @@ class HubRevisionTest(unittest.TestCase):
with mock.patch.dict(os.environ, self.modified_environ, clear=True):
self.prepare_repo_data() # no tag, default get master
with tempfile.TemporaryDirectory() as temp_cache_dir:
time.sleep(11)
snapshot_path = snapshot_download(
self.model_id, cache_dir=temp_cache_dir)
assert os.path.exists(
@@ -94,6 +95,7 @@ class HubRevisionTest(unittest.TestCase):
branch_name = 'test'
self.add_new_file_and_branch_to_repo(branch_name)
with tempfile.TemporaryDirectory() as temp_cache_dir:
time.sleep(10)
snapshot_path = snapshot_download(
self.model_id,
revision=branch_name,
@@ -101,6 +103,7 @@ class HubRevisionTest(unittest.TestCase):
assert os.path.exists(
os.path.join(snapshot_path, download_model_file_name))
with tempfile.TemporaryDirectory() as temp_cache_dir:
time.sleep(11)
file_path = model_file_download(
self.model_id,
download_model_file_name,
@@ -113,7 +116,7 @@ class HubRevisionTest(unittest.TestCase):
self.prepare_repo_data_and_tag()
t1 = datetime.now().isoformat(sep=' ', timespec='seconds')
logger.info('First time: %s' % t1)
time.sleep(10)
time.sleep(11)
self.add_new_file_and_tag_to_repo()
t2 = datetime.now().isoformat(sep=' ', timespec='seconds')
logger.info('Second time: %s' % t2)
@@ -134,6 +137,7 @@ class HubRevisionTest(unittest.TestCase):
version.__release_datetime__ = t2
logger.info('Setting __release_datetime__ to: %s' % t2)
with tempfile.TemporaryDirectory() as temp_cache_dir:
time.sleep(11)
snapshot_path = snapshot_download(
self.model_id, cache_dir=temp_cache_dir)
assert os.path.exists(
@@ -169,6 +173,7 @@ class HubRevisionTest(unittest.TestCase):
assert not os.path.exists(
os.path.join(snapshot_path, download_model_file_name2))
with tempfile.TemporaryDirectory() as temp_cache_dir:
time.sleep(11)
snapshot_path = snapshot_download(
self.model_id,
revision=self.revision2,