From 8e28c327fc832f7c6acffa2870c27db01658cc43 Mon Sep 17 00:00:00 2001 From: Harutaka Kawamura Date: Tue, 27 Oct 2020 23:34:51 +0900 Subject: [PATCH] Fix assertion error message for MLflowCallback (#8091) --- 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 e11eeb78ada..db9a81c0fa3 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