update

liding-nv · liding-nv · commit fe27b3482ee2 · 2026-05-31T19:32:52.000-07:00
Signed-off-by: Li Ding &lt;liding@nvidia.com&gt;
diff --git a/tests/unit_tests/models/test_mimo_model.py b/tests/unit_tests/models/test_mimo_model.py
@@ -798,9 +798,7 @@ def _stub_mimo_config(
         return SimpleNamespace(
             language_model_spec=SimpleNamespace(params={'config': language_config}),
             modality_submodules_spec={
-                "images": SimpleNamespace(
-                    params={} if modality_params is None else modality_params
-                )
+                "images": SimpleNamespace(params={} if modality_params is None else modality_params)
             },
         )
 
@@ -830,9 +828,7 @@ def test_empty_encoder_output_uses_language_dtype_without_modality_config(self,
         monkeypatch.setattr(torch.cuda, "current_device", lambda: torch.device("cpu"))
         model = MimoModel.__new__(MimoModel)
         model.mimo_config = self._stub_mimo_config(
-            hidden_size=8,
-            language_dtype=torch.bfloat16,
-            modality_params={},
+            hidden_size=8, language_dtype=torch.bfloat16, modality_params={}
         )
 
         output = model._empty_encoder_output("images")
@@ -846,9 +842,7 @@ def test_empty_encoder_output_defaults_to_float32_without_language_dtype(self, m
         monkeypatch.setattr(torch.cuda, "current_device", lambda: torch.device("cpu"))
         model = MimoModel.__new__(MimoModel)
         model.mimo_config = self._stub_mimo_config(
-            hidden_size=8,
-            include_language_dtype=False,
-            modality_params={},
+            hidden_size=8, include_language_dtype=False, modality_params={}
         )
 
         output = model._empty_encoder_output("images")
