fix requirements

modelscope_agent/tools/code_interpreter/code_interpreter_nb.py

@@ -12,19 +12,20 @@
 import json
 import json5
 import nbformat
-from jupyter_client import BlockingKernelClient
 from modelscope_agent.tools.base import BaseTool, register_tool
 from modelscope_agent.utils.utils import extract_code
-from nbclient import NotebookClient
-from nbclient.exceptions import CellTimeoutError, DeadKernelError
-from nbformat import NotebookNode
-from nbformat.v4 import new_code_cell, new_markdown_cell, new_output
-from rich.box import MINIMAL
-from rich.console import Console, Group
-from rich.live import Live
-from rich.markdown import Markdown
-from rich.panel import Panel
-from rich.syntax import Syntax
+
+try:
+ from nbclient import NotebookClient
+ from nbclient.exceptions import CellTimeoutError, DeadKernelError
+ from nbformat import NotebookNode
+ from nbformat.v4 import new_code_cell, new_markdown_cell, new_output
+ from rich.console import Console
+ from rich.syntax import Syntax
+except ImportError:
+ raise ImportError(
+ 'Please install nbclient, nbformat, rich by running `pip install nbclient nbformat rich`'
+ )
 
 
 @register_tool('code_interpreter')

modelscope_agent/tools/metagpt_tools/libs/data_preprocess.py

@@ -6,10 +6,16 @@
 import numpy as np
 import pandas as pd
 from modelscope_agent.tools.metagpt_tools.tool_registry import register_tool
-from sklearn.impute import SimpleImputer
-from sklearn.preprocessing import (LabelEncoder, MaxAbsScaler, MinMaxScaler,
- OneHotEncoder, OrdinalEncoder, RobustScaler,
- StandardScaler)
+
+try:
+ from sklearn.impute import SimpleImputer
+ from sklearn.preprocessing import (LabelEncoder, MaxAbsScaler,
+ MinMaxScaler, OneHotEncoder,
+ OrdinalEncoder, RobustScaler,
+ StandardScaler)
+except ImportError:
+ raise ImportError(
+ 'Please install scikit-learn by running `pip install scikit-learn`')
 
 TAGS = ['data preprocessing', 'machine learning']
 

modelscope_agent/tools/metagpt_tools/libs/feature_engineering.py

@@ -8,10 +8,15 @@
 from joblib import Parallel, delayed
 from modelscope_agent.tools.metagpt_tools.libs.data_preprocess import MLProcess
 from modelscope_agent.tools.metagpt_tools.tool_registry import register_tool
-from pandas.core.dtypes.common import is_object_dtype
-from sklearn.feature_selection import VarianceThreshold
-from sklearn.model_selection import KFold
-from sklearn.preprocessing import KBinsDiscretizer, PolynomialFeatures
+
+try:
+ from pandas.core.dtypes.common import is_object_dtype
+ from sklearn.feature_selection import VarianceThreshold
+ from sklearn.model_selection import KFold
+ from sklearn.preprocessing import KBinsDiscretizer, PolynomialFeatures
+except ImportError:
+ raise ImportError(
+ 'Please install scikit-learn by running `pip install scikit-learn`')
 
 TAGS = ['feature engineering', 'machine learning']