From 614c817ccb5eeb0e052c487dde2663ac0b5b23ee Mon Sep 17 00:00:00 2001
From: harupy <17039389+harupy@users.noreply.github.com>
Date: Tue, 27 Oct 2020 21:03:10 +0900
Subject: [PATCH] Fix assertion error message for MLflowCallback

---
 src/transformers/integrations.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/transformers/integrations.py b/src/transformers/integrations.py
index e11eeb78adabbe..db9a81c0fa38e5 100644
--- a/src/transformers/integrations.py
+++ b/src/transformers/integrations.py
@@ -432,7 +432,7 @@ class MLflowCallback(TrainerCallback):
     MAX_LOG_SIZE = 100
 
     def __init__(self):
-        assert _has_mlflow, "MLflow requires mlflow to be installed. Run `pip install mlflow`."
+        assert _has_mlflow, "MLflowCallback requires mlflow to be installed. Run `pip install mlflow`."
         self._initialized = False
         self._log_artifacts = False