Skip to content

Commit

Permalink
Removed SkLearn-Pandas from package dependencies
Browse files Browse the repository at this point in the history
  • Loading branch information
vruusmann committed Mar 14, 2024
1 parent 132f6d7 commit 374e833
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 5 deletions.
3 changes: 1 addition & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,6 @@
install_requires = [
"dill>=0.3.4",
"joblib>=0.13.0",
"scikit-learn>=1.0",
"sklearn-pandas>=1.8.0"
"scikit-learn>=1.0"
]
)
8 changes: 5 additions & 3 deletions sklearn2pmml/__init__.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,14 @@
from pandas import CategoricalDtype
try:
from sklearn_pandas import DataFrameMapper
except ImportError:
class DataFrameMapper(object):
pass
from sklearn.base import BaseEstimator
from sklearn.compose import ColumnTransformer
from sklearn.exceptions import NotFittedError
from sklearn.feature_selection import SelectFromModel, SelectorMixin
from sklearn.pipeline import FeatureUnion, Pipeline
from sklearn_pandas import DataFrameMapper
from sklearn2pmml.resources import _package_classpath
from subprocess import PIPE, Popen
from zipfile import ZipFile
Expand All @@ -17,7 +21,6 @@
import platform
import re
import sklearn
import sklearn_pandas
import tempfile
import warnings

Expand Down Expand Up @@ -269,7 +272,6 @@ def sklearn2pmml(estimator, pmml_path, with_repr = False, java_home = None, java
print("python: {0}".format(platform.python_version()))
print("sklearn2pmml: {0}".format(__version__))
print("sklearn: {0}".format(sklearn.__version__))
print("sklearn_pandas: {0}".format(sklearn_pandas.__version__))
print("pandas: {0}".format(pandas.__version__))
print("numpy: {0}".format(numpy.__version__))
print("dill: {0}".format(dill.__version__))
Expand Down

0 comments on commit 374e833

Please sign in to comment.