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

style: fix yapf/flake8 formatting in hf util tests

Browse Source 
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
suluyan
2026-07-22 16:33:30 +08:00
parent 5aef4de5e4
commit 7e9a2e67aa
1 changed files with 11 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/utils/test_hf_util.py
View File

@@ -372,8 +372,7 @@ class HFUtilTest(unittest.TestCase):
# item assignment.
_get_class_from_dynamic_module('modeling.Foo', remote_id)
sd.assert_called_once_with(
remote_id, local_files_only=False)
sd.assert_called_once_with(remote_id, local_files_only=False)
self.assertEqual(captured['class_reference'], 'modeling.Foo')
self.assertEqual(captured['pretrained'], downloaded)
@@ -453,25 +452,23 @@ class HFUtilTest(unittest.TestCase):
self.assertEqual(
_ms_download_kwargs_from_hf({}), {'local_files_only': False})
self.assertEqual(
_ms_download_kwargs_from_hf(
{
'local_files_only': True,
'cache_dir': '/tmp/c',
'token': 'sekrit',
'token_ignored': True,
},
revision='main'),
got = _ms_download_kwargs_from_hf(
{
'local_files_only': True,
'cache_dir': '/tmp/c',
'token': 'sekrit',
'token_ignored': True,
},
revision='main')
self.assertEqual(
got, {
'local_files_only': True,
'cache_dir': '/tmp/c',
'token': 'sekrit',
'revision': 'master',
})
# HF token=True means "default creds"; only string tokens are forwarded.
self.assertNotIn(
'token',
_ms_download_kwargs_from_hf({'token': True}))
self.assertNotIn('token', _ms_download_kwargs_from_hf({'token': True}))
def test_get_model_dir_forwards_cache_dir_and_token(self):
"""from_pretrained download path must forward cache_dir and token."""