# Contributor: Carlos Giraldo <cgiraldo@gradiant.org>
# Maintainer: Carlos Giraldo <cgiraldo@gradiant.org>
pkgname=py3-scikit-learn
pkgver=1.1.3
pkgrel=1
pkgdesc="a Python module for machine learning built on top of SciPy"
url="https://scikit-learn.org/"
arch="all !s390x" # blocked by py3-joblib
license="BSD-3-Clause"
options="!check" # no test suite
depends="py3-numpy py3-scipy py3-joblib py3-threadpoolctl"
makedepends="cython py3-setuptools py3-numpy-dev python3-dev"
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() {
	export SETUPTOOLS_USE_DISTUTILS=stdlib
	python3 setup.py build_ext -j $JOBS
}

package() {
	export SETUPTOOLS_USE_DISTUTILS=stdlib
	python3 setup.py install --skip-build --root="$pkgdir"
}

sha512sums="
4c3b094812653fa4f2c9db510bb1dac9129700da53282f0265a528a890ea2e2f9318d5a35c7100dcf26e91ecc286a735e7118b34948c1370ad1ce3ee83fc7778  scikit-learn-1.1.3.tar.gz
"