From 7e9a2e67aadd2f4968561c6af1f9a28c2d3fc362 Mon Sep 17 00:00:00 2001
From: suluyan <suluyan.sly@aliabab-inc.com>
Date: Wed, 22 Jul 2026 16:33:30 +0800
Subject: [PATCH] style: fix yapf/flake8 formatting in hf util tests

Co-authored-by: Cursor <cursoragent@cursor.com>
---
 tests/utils/test_hf_util.py | 25 +++++++++++--------------
 1 file changed, 11 insertions(+), 14 deletions(-)

diff --git a/tests/utils/test_hf_util.py b/tests/utils/test_hf_util.py
index 044ba436..e47f8b56 100644
--- a/tests/utils/test_hf_util.py
+++ b/tests/utils/test_hf_util.py
@@ -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."""