mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-16 16:27:45 +01:00
update test
This commit is contained in:
@@ -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}
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user