update test

This commit is contained in:
Yunnglin
2025-11-07 16:52:26 +08:00
parent 6d399f2da6
commit d6e02eb2ce
2 changed files with 4 additions and 1 deletions

View File

@@ -22,7 +22,7 @@ do
echo "get gpu lock $gpu"
CONTAINER_NAME="modelscope-ci-$idx"
let is_get_file_lock=true
is_get_file_lock=true
# pull image if there are update
docker pull ${IMAGE_NAME}:${IMAGE_VERSION}

View File

@@ -3,6 +3,7 @@ import unittest
from modelscope import HubApi
from modelscope.utils.logger import get_logger
from modelscope.utils.test_utils import test_level
logger = get_logger()
@@ -14,11 +15,13 @@ class HubListHubTest(unittest.TestCase):
def setUp(self):
self.api = HubApi()
@unittest.skipUnless(test_level() >= 3, 'skip test in current test level')
def test_list_datasets(self):
# Use default args
result = self.api.list_datasets(owner_or_group=default_owner)
logger.info(f'List datasets result: {result}')
@unittest.skipUnless(test_level() >= 3, 'skip test in current test level')
def test_list_datasets_with_args(self):
result = self.api.list_datasets(
owner_or_group=default_owner,