From 996b7cdbe8258dc1ea21fca4c7b67d8d60539f22 Mon Sep 17 00:00:00 2001 From: Ivo Alxneit Date: Fri, 18 Mar 2022 09:28:20 +0100 Subject: [PATCH] BUGFIX: sklearn does not provide six anymore. Import standalone package. Also add six to requirements Signed-off-by: Ivo Alxneit --- 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 16c8971..c726626 100644 --- a/mlrose/neural.py +++ b/mlrose/neural.py @@ -9,7 +9,7 @@ from abc import abstractmethod 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 9e8eec2..c29ca5b 100644 --- a/setup.py +++ b/setup.py @@ -36,6 +36,6 @@ setup(name='mlrose', "Topic :: Software Development :: Libraries", "Topic :: Software Development :: Libraries :: Python Modules"], packages=['mlrose'], - install_requires=['numpy', 'scipy', 'sklearn'], + install_requires=['numpy', 'scipy', 'sklearn', 'six'], python_requires='>=3', zip_safe=False) -- 2.25.1