From 5dc32d5c24de9bf036745244aa0645af35a71617 Mon Sep 17 00:00:00 2001 From: Prashanth Aditya Susarla Date: Mon, 13 Feb 2023 19:09:54 +0530 Subject: [PATCH] sklearn.externals.six was removed in sklearn 0.23, importing six directly after setting it up as a project dependency --- mlrose/neural.py | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mlrose/neural.py b/mlrose/neural.py index 16c8971a..c726626f 100644 --- a/mlrose/neural.py +++ b/mlrose/neural.py @@ -9,7 +9,7 @@ import numpy as np from sklearn.base import BaseEstimator, ClassifierMixin, RegressorMixin from sklearn.metrics import mean_squared_error, log_loss -from sklearn.externals import six +import six from .activation import identity, relu, sigmoid, softmax, tanh from .algorithms import random_hill_climb, simulated_annealing, genetic_alg from .opt_probs import ContinuousOpt diff --git a/setup.py b/setup.py index 3c010a8e..cbcada8a 100644 --- a/setup.py +++ b/setup.py @@ -36,6 +36,6 @@ def readme(): "Topic :: Software Development :: Libraries", "Topic :: Software Development :: Libraries :: Python Modules"], packages=['mlrose'], - install_requires=['numpy', 'scipy', 'scikit-learn'], + install_requires=['six', 'numpy', 'scipy', 'scikit-learn'], python_requires='>=3', zip_safe=False)