Non-installed authorized modules: scikit-learn

[lfirek]

Hello,

I had a problem when I started using Plexe in my environment. When I tried to build a model from the examples in the repository, I got an error. I tried downgrading the version of Plexe, but the problem remained.

Can anyone suggest what I should do in this situation?

Environment:

• OS: MacOS
• Python Version 3.11.7
• PIP 25.1.1
• scikit-learn 1.6.1
• plexe 0.26.2

Starting model build...
[2025-07-14 22:56:50,263 - plexe.core.object_registry - INFO - (MainThread)]: - Registry: registered <class 'bool'>://input_schema_is_locked (immutable=True, total: 1 items)
[2025-07-14 22:56:50,263 - plexe.core.object_registry - INFO - (MainThread)]: - Registry: registered <class 'bool'>://output_schema_is_locked (immutable=True, total: 2 items)
[2025-07-14 22:56:50,263 - plexe.core.object_registry - INFO - (MainThread)]: - Registry: registered <class 'plexe.callbacks.Callback'>://0 (immutable=False, total: 3 items)
[2025-07-14 22:56:50,263 - plexe.core.object_registry - INFO - (MainThread)]: - Registry: registered <class 'plexe.callbacks.Callback'>://1 (immutable=False, total: 4 items)
[2025-07-14 22:56:50,263 - plexe.core.object_registry - INFO - (MainThread)]: - Registry: registered <class 'plexe.internal.common.datasets.interface.TabularConvertible'>://dataset_0 (immutable=True, total: 5 items)
[2025-07-14 22:56:50,334 - plexe.internal.models.callbacks.mlflow - INFO - (MainThread)]: - Started parent run 'f3526a28d57246f9ae276122b9535ae3' in experiment 'house-prices-20250714-225650'
System · Step 1 · 22:56:50
🚀 Starting model build for intent: Predict house prices based on square foo...
W0714 22:56:54.059000 7815 site-packages/torch/distributed/elastic/multiprocessing/redirects.py:29] NOTE: Redirects are currently not supported in Windows or MacOs.
[2025-07-14 22:56:54,128 - plexe.model_builder - ERROR - (MainThread)]: - Error during model building: Non-installed authorized modules: scikit-learn. Please install these modules or remove them from the authorized imports list.
Traceback (most recent call last):
  File "/plexe-tests/transaction-predict-v1.py", line 45, in <module>
    model.build(
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/deprecated/classic.py", line 199, in wrapper_function
    return wrapped_(*args_, **kwargs_)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/plexe/models.py", line 194, in build
    built_model = builder.build(
                  ^^^^^^^^^^^^^^
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/plexe/model_builder.py", line 258, in build
    raise e
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/plexe/model_builder.py", line 154, in build
    agent = PlexeAgent(
            ^^^^^^^^^^^
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/plexe/agents/agents.py", line 120, in __init__
    self.eda_agent = EdaAgent(
                     ^^^^^^^^^
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/plexe/agents/dataset_analyser.py", line 51, in __init__
    self.agent = CodeAgent(
                 ^^^^^^^^^^
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/smolagents/agents.py", line 1552, in __init__
    self.python_executor = self.create_python_executor()
                           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/smolagents/agents.py", line 1567, in create_python_executor
    return LocalPythonExecutor(
           ^^^^^^^^^^^^^^^^^^^^
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/smolagents/local_python_executor.py", line 1596, in __init__
    self._check_authorized_imports_are_installed()
  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/smolagents/local_python_executor.py", line 1615, in _check_authorized_imports_are_installed
    raise InterpreterError(
smolagents.local_python_executor.InterpreterError: Non-installed authorized modules: scikit-learn. Please install these modules or remove them from the authorized imports list.
🏃 View run model-2025-0...-20250714-225650 at: http://127.0.0.1:5000/#/experiments/168489108307540325/runs/f3526a28d57246f9ae276122b9535ae3
🧪 View experiment at: http://127.0.0.1:5000/#/experiments/168489108307540325

[vusynhathoang]

same

[dmater01]

Same Here 👍

Pexe's Allowed packages: ['pandas', 'numpy', 'scikit-learn', 'sklearn', 'joblib', 'mlxtend', 'xgboost', 'pyarrow', 'statsmodels', 'torch', 'transformers', 'tokenizers', 'accelerate', 'safetensors']
🏥 STROKE RISK PREDICTION MODEL

✅ OpenAI API key found
📊 Loading stroke dataset...
✅ Dataset loaded successfully!
📈 Total patients: 5,110
📈 Stroke cases: 249
📈 Stroke rate: 4.9%
📋 Dataset overview:
Age range: 0 - 82 years
BMI range: 10.3 - 97.6
Glucose range: 55 - 272

📋 Data preview:
id gender age hypertension heart_disease ... Residence_type avg_glucose_level bmi smoking_status stroke
0 9046 Male 67.0 0 1 ... Urban 228.69 36.6 formerly smoked 1
1 51676 Female 61.0 0 0 ... Rural 202.21 NaN never smoked 1
2 31112 Male 80.0 0 1 ... Rural 105.92 32.5 never smoked 1

[3 rows x 12 columns]

🎯 Medical Intent:
'predict stroke occurrence based on age, gender, hypertension, heart_disease, ever_married, work_type, Residence_type, avg_glucose_level, bmi, and smoking_status'

🤖 Creating medical AI model...
✅ Plex model object created successfully!

🚀 Ready to build your stroke prediction model? This will cost ~$15-25 (y/n): y

🎬 AI agents are now collaborating...
🕒 Expected duration: ~1 hour
💰 Expected cost: ~$15-25
🤖 AI agents will handle:
• Data validation and cleaning
• Feature engineering for medical data
• Algorithm selection and training
• Model evaluation and optimization

🔧 Building model with correct parameters...

🔧 Building model with correct parameters...
[2025-07-24 10:14:00,667 - plexe.core.object_registry - INFO - (MainThread)]: - Registry: registered <class 'bool'>://input_schema_is_locked (immutable=True, total: 1 items)
[2025-07-24 10:14:00,667 - plexe.core.object_registry - INFO - (MainThread)]: - Registry: registered <class 'bool'>://output_schema_is_locked (immutable=True, total: 2 items)
[2025-07-24 10:14:00,667 - plexe.core.object_registry - INFO - (MainThread)]: - Registry: registered <class 'plexe.callbacks.Callback'>://0 (immutable=False, total: 3 items)
[2025-07-24 10:14:00,668 - plexe.core.object_registry - INFO - (MainThread)]: - Registry: registered <class 'plexe.internal.common.datasets.interface.TabularConvertible'>://dataset_0 (immutable=True, total: 4 items)
System · Step 1 · 10:14:00
🚀 Starting model build for intent: predict stroke occurrence based on age, ...
[2025-07-24 10:14:00,720 - plexe.model_builder - ERROR - (MainThread)]: - Error during model building: Non-installed authorized modules: scikit-learn. Please install these modules or remove them from the authorized imports list.

❌ Error during model building: Non-installed authorized modules: scikit-learn. Please install these modules or remove them from the authorized imports list.
💡 This might be due to:
• OpenAI API connectivity issues
• Insufficient OpenAI credits
• Data format issues
• Temporary service interruptions

[dmater01]

All issue have been verified :
• OpenAI API connectivity issues --working
• Insufficient OpenAI credits -- working
• Data format issues -- none
• Temporary service interruptions -- working

[dmater01]

To Plexe support , please find the information gathered:

The exact error: "Non-installed authorized modules: scikit-learn."

Your Python version.

The versions of plexe (0.26.2) and scikit-learn (1.6.1).

The output confirming scikit-learn is in plexe.config.config.code_generation.allowed_packages.

Confirmation that your OpenAI API is working correctly.

[vaibs-d]

Hi @lfirek and @dmater01 thanks a lot for raising this. I'm looking into this now!

[kyyjoensuu]

Fixed personally by deleting the "scikit-learn" lines in the lists in the following files. Make sure to leave the "sklearn" lines

.venv/lib/python3.13/site-packages/plexe/config.py
.venv/lib/python3.13/site-packages/plexe/agents/feature_engineer.py