# Contributor: Carlos Giraldo <cgiraldo@gradiant.org>
# Maintainer: Carlos Giraldo <cgiraldo@gradiant.org>
pkgname=py3-scikit-learn
pkgver=1.5.2
pkgrel=0
pkgdesc="a Python module for machine learning built on top of SciPy"
url="https://scikit-learn.org/"
arch="all"
license="BSD-3-Clause"
options="!check" # no test suite
depends="
	py3-joblib
	py3-numpy
	py3-scipy
	py3-threadpoolctl
	python3
	"
makedepends="
	cython
	py3-gpep517
	py3-meson-python
	py3-numpy-dev
	py3-setuptools
	py3-wheel
	python3-dev
	"
subpackages="$pkgname-pyc"
source="https://files.pythonhosted.org/packages/source/s/scikit-learn/scikit_learn-$pkgver.tar.gz"
builddir="$srcdir/scikit_learn-$pkgver"

case "$CARCH" in
aarch64)
	export SKLEARN_NO_OPENMP=1
	;;
esac

build() {
	CFLAGS="$CFLAGS -O2 -flto=auto" \
	gpep517 build-wheel \
		--wheel-dir .dist \
		--output-fd 3 3>&1 >&2
}

package() {
	python3 -m installer -d "$pkgdir" \
		.dist/*.whl

	find "$pkgdir" -type d -name "tests" -exec rm -rf {} \+
}

sha512sums="
f8e31c733d435e0fb879ae84f253ef036a1586516814350ac752e2133b4ef958e9ae3f3e2b9fd38759bae543a8c767653370f49785014bb47b9767c84cc247bd  scikit_learn-1.5.2.tar.gz
"
