1 files changed, 12 insertions(+), 13 deletions(-)

M 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