From 30b69279223a89b41c1b1be5e34f463da76f4089 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Fri, 1 Nov 2024 13:00:04 -0700 Subject: [PATCH] type ignore --- llmfoundry/callbacks/hf_checkpointer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py index 8abe845f1b..688d8deb74 100644 --- 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: