From 5565dcdd354f24a118befebb8673bf58dc2f0f51 Mon Sep 17 00:00:00 2001 From: Aymeric Augustin Date: Mon, 23 Dec 2019 21:16:26 +0100 Subject: [PATCH] Remove warning when scikit-learn isn't available. Most users don't need it. --- src/transformers/data/metrics/__init__.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/src/transformers/data/metrics/__init__.py b/src/transformers/data/metrics/__init__.py index 4d8d55a1c7..f65c76faeb 100644 --- a/src/transformers/data/metrics/__init__.py +++ b/src/transformers/data/metrics/__init__.py @@ -14,18 +14,12 @@ # See the License for the specific language governing permissions and # limitations under the License. -import logging - - -logger = logging.getLogger(__name__) - try: from scipy.stats import pearsonr, spearmanr from sklearn.metrics import matthews_corrcoef, f1_score _has_sklearn = True except (AttributeError, ImportError) as e: - logger.warning("To use data.metrics please install scikit-learn. See https://scikit-learn.org/stable/index.html") _has_sklearn = False