From d0957ebaecd374ac6801cde72d0abca7f4891801 Mon Sep 17 00:00:00 2001 From: ruanhaitao Date: Wed, 31 Jul 2024 03:02:15 +0000 Subject: [PATCH] fix import error message for 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..2dfd9a51 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_path needs_to_be_installed.add((library_import_name, library_import_path)) if needs_to_be_installed: raise ImportError(