intel/scikit-learn-intelex

Unable to run patch_sklearn()

briankw1108 opened this issue · 2 comments

briankw1108 commented

Describe the bug
I updated it to the latest version of 2024.1.0, but I got an error when I tried to run patch_sklearn() after importing patch_sklearn from sklearnex.

The error message says that: AttributeError: module 'threadpoolctl' has no attribute 'ThreadpoolController'

To Reproduce
Steps to reproduce the behavior:

  1. pip install scikit-learn-intelex
  2. from sklearnex import patch_sklearn
  3. run patch_sklearn()
  4. AttributeError: module 'threadpoolctl' has no attribute 'ThreadpoolController'

Expected behavior
Able to enable patching

Output/Screenshots
image

Environment:

  • Databricks Runtime Version 13.3
  • Python 3.10.12
Alexsandruss commented

threadpoolctl is dependency of scikit-learn and scikit-learn-intelex and definitely contains ThreadpoolController.
This issue is highly likely on environment side.