type ignore

mosaicml · Nov 1, 2024 · 30b6927 · 30b6927
1 parent 7a423b1
commit 30b6927
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -195,7 +195,7 @@ def save_model_patch(*args: Any, **kwargs: Any):
                 e,
             )
 
-    mlflow.transformers.save_model = save_model_patch
+    mlflow.transformers.save_model = save_model_patch  # type: ignore
 
     mlflow.set_tracking_uri(mlflow_logger.tracking_uri)
     if mlflow_logger.model_registry_uri is not None: