aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--gnu/packages/machine-learning.scm25
1 files changed, 12 insertions, 13 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm
index b5bf9aa5fe..7d833005bc 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -1663,39 +1663,38 @@ and forecasting.")
(define-public python-imbalanced-learn
(package
(name "python-imbalanced-learn")
- (version "0.9.1")
+ (version "0.11.0")
(source
(origin
(method url-fetch)
(uri (pypi-uri "imbalanced-learn" version))
(sha256
- (base32 "0qnrmysnqpc8ii1w5n8mci20gcjhmjr7khvk7f2apdbqc2pgf52f"))))
+ (base32 "1p4gdgc8nsq0vjmw4y4d2bp9g0m1c23d0zgrzs90pnz6b24ax0km"))))
(build-system pyproject-build-system)
(arguments
(list
+ #:test-flags '(list "-k"
+ ;; Although we cannot satify the Tensorflow and Keras requirements
+ ;; (python-keras >= 2.4.3 and tensorflow >= 2.4.3), all tests
+ ;; besides these pass.
+ "not balanced_batch_generator and not BalancedBatchGenerator")
#:phases '(modify-phases %standard-phases
(add-after 'unpack 'unbreak-tests
(lambda _
- ;; The doctests require tensorflow
- (substitute* "setup.cfg"
- (("--doctest-modules")
- ""))
;; Some tests require a home directory
(setenv "HOME"
- (getcwd))
- ;; We don't have keras
- (delete-file "imblearn/keras/tests/test_generator.py")
- ;; We don't have tensorflow
- (delete-file
- "imblearn/tensorflow/tests/test_generator.py"))))))
+ (getcwd)))))))
(propagated-inputs (list python-joblib python-numpy python-scikit-learn
python-scipy python-threadpoolctl))
(native-inputs (list python-black
python-flake8
+ python-keras
python-mypy
+ python-numpydoc
python-pandas
python-pytest
- python-pytest-cov))
+ python-pytest-cov
+ tensorflow))
(home-page "https://github.com/scikit-learn-contrib/imbalanced-learn")
(synopsis "Toolbox for imbalanced dataset in machine learning")
(description