update to 0.9.0

2 files changed, 32 insertions, 19 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 50f034d0db69..7bf692fd7130 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,19 +1,21 @@
 pkgbase = python-imbalanced-learn
-	pkgdesc = imbalanced-learn is a python package offering a number of re-sampling techniques commonly used in datasets showing strong between-class imbalance.
-	pkgver = 0.7.0
+	pkgdesc = Toolbox for imbalanced dataset in machine learning
+	pkgver = 0.9.0
 	pkgrel = 1
 	url = https://github.com/scikit-learn-contrib/imbalanced-learn
 	arch = any
 	license = MIT
 	makedepends = python-setuptools
-	depends = python
+	makedepends = python-build
+	makedepends = python-installer
+	makedepends = python-wheel
+	depends = python>=3.7
 	depends = python-scipy
 	depends = python-numpy
 	depends = python-scikit-learn
 	optdepends = python-keras
 	optdepends = python-tensorflow
-	source = https://files.pythonhosted.org/packages/3e/c1/2a1b332589b3e668a560c0cb00d98c204ece60f178a0dbde4e2165194ce7/imbalanced-learn-0.7.0.tar.gz
-	sha256sums = da59de0d1c0fa66f62054dd9a0a295a182563aa1abbb3bf9224a3678fcfe8fa4
+	source = python-imbalanced-learn-0.9.0.tar.gz::https://files.pythonhosted.org/packages/source/i/imbalanced-learn/imbalanced-learn-0.9.0.tar.gz
+	sha256sums = 836a4c137cc3c10310d4f6cd5ec34600ff488d7f8c243a997c3f9b551c91d0b2
 
 pkgname = python-imbalanced-learn
-
diff --git a/PKGBUILD b/PKGBUILD
index 6c8b96968cb3..3d74cb2225ac 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,25 +1,36 @@
-# Maintainer: Federico Cassani <federico dot cassani at outlook dot com> 
+# Maintainer: Luis Martinez <luis dot martinez at disroot dot org>
+# Contributor: Federico Cassani <federico dot cassani at outlook dot com>
 
-pkgname="python-imbalanced-learn"
-_pkgname="imbalanced-learn"
-pkgver=0.7.0
+pkgname=python-imbalanced-learn
+_pkg="${pkgname#python-}"
+pkgver=0.9.0
 pkgrel=1
-pkgdesc="imbalanced-learn is a python package offering a number of re-sampling techniques commonly used in datasets showing strong between-class imbalance."
+pkgdesc='Toolbox for imbalanced dataset in machine learning'
 arch=('any')
 url="https://github.com/scikit-learn-contrib/imbalanced-learn"
 license=('MIT')
-depends=('python' 'python-scipy' 'python-numpy' 'python-scikit-learn')
+depends=('python>=3.7' 'python-scipy' 'python-numpy' 'python-scikit-learn')
 optdepends=('python-keras' 'python-tensorflow')
-makedepends=('python-setuptools')
-source=("https://files.pythonhosted.org/packages/3e/c1/2a1b332589b3e668a560c0cb00d98c204ece60f178a0dbde4e2165194ce7/${_pkgname}-${pkgver}.tar.gz")
-sha256sums=('da59de0d1c0fa66f62054dd9a0a295a182563aa1abbb3bf9224a3678fcfe8fa4')
+makedepends=('python-setuptools' 'python-build' 'python-installer' 'python-wheel')
+source=("$pkgname-$pkgver.tar.gz::https://files.pythonhosted.org/packages/source/${_pkg::1}/$_pkg/$_pkg-$pkgver.tar.gz")
+sha256sums=('836a4c137cc3c10310d4f6cd5ec34600ff488d7f8c243a997c3f9b551c91d0b2')
+
+prepare() {
+	cd "$_pkg-$pkgver"
+	sed -i "/packages=/s/()/(exclude=['*tests*'])/" setup.py
+}
 
 build() {
-    cd $srcdir/$_pkgname-$pkgver
-    python setup.py build
+	cd "$_pkg-$pkgver"
+	python -m build --wheel --no-isolation
 }
 
 package() {
-    cd $srcdir/$_pkgname-$pkgver
-    python setup.py install --root="$pkgdir" --optimize=1 --skip-build
+	cd "$_pkg-$pkgver"
+	PYTHONHASHSEED=0 python -m installer --destdir="$pkgdir/" dist/*.whl
+	local _site="$(python -c 'import site; print(site.getsitepackages()[0])')"
+	install -d "$pkgdir/usr/share/licenses/$pkgname/"
+	ln -s \
+		"$_site/${_pkg/-/_}-$pkgver.dist-info/LICENSE" \
+		"$pkgdir/usr/share/licenses/$pkgname/"
 }