From 3447debe8ac0d8c64debf780d199fe9aff616653 Mon Sep 17 00:00:00 2001 From: ArkiZh <1797466468@qq.com> Date: Wed, 14 Aug 2024 18:43:07 +0800 Subject: [PATCH] Fix: message for lack of sklearn When relying on `sklearn`, the help message that should be printed is `pip install scikit-learn`, not `pip install sklearn`. --- src/evaluate/loading.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/evaluate/loading.py b/src/evaluate/loading.py index a9cbbe60..2ffd5b51 100644 --- a/src/evaluate/loading.py +++ b/src/evaluate/loading.py @@ -259,7 +259,7 @@ def _download_additional_modules( try: lib = importlib.import_module(library_import_name) # noqa F841 except ImportError: - library_import_name = "scikit-learn" if library_import_name == "sklearn" else library_import_name + library_import_path = "scikit-learn" if library_import_name == "sklearn" else library_import_name needs_to_be_installed.add((library_import_name, library_import_path)) if needs_to_be_installed: raise ImportError(