apply format check suggestion

citadel-ai · Feb 19, 2024 · 64f7e95 · 64f7e95
1 parent daeb706
commit 64f7e95
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/metrics/model_manager/test_model_manager.py b/tests/metrics/model_manager/test_model_manager.py
@@ -61,8 +61,9 @@ def test_model_manager_initiation(mock_model_manager):
 
 
 def test_model_manager_fetch_model(mock_model_manager):
-    with patch.dict(
-        'langcheck.metrics.model_manager._model_management.LOADER_MAP',
-        {'load_auto_model_for_text_classification': MagicMock()}):
+    with \
+        patch.dict(
+            'langcheck.metrics.model_manager._model_management.LOADER_MAP',
+            {'load_auto_model_for_text_classification': MagicMock()}):
         model = mock_model_manager.fetch_model(language='zh', metric='toxicity')
         assert model is not None