commit change

mosaicml · Jun 17, 2024 · 9f00f61 · 9f00f61
1 parent 412e0cd
commit 9f00f61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/a_scripts/inference/test_convert_composer_to_hf.py b/tests/a_scripts/inference/test_convert_composer_to_hf.py
@@ -928,7 +928,7 @@ def test_huggingface_conversion_callback(
 
     context_manager = te.onnx_export(
         True,
-    ) if trainer_precision == 'amp_fp8' else contextlib.nullcontext()
+    ) if trainer_precision == 'amp_fp8' else contextlib.nullcontext()  # type: ignore
     with context_manager:
         with FSDP.summon_full_params(
             trainer.state.model,