diff options
-rw-r--r-- | .SRCINFO | 17 | ||||
-rw-r--r-- | .gitignore | 4 | ||||
-rw-r--r-- | PKGBUILD | 67 |
3 files changed, 88 insertions, 0 deletions
diff --git a/.SRCINFO b/.SRCINFO new file mode 100644 index 000000000000..d963e8583bfd --- /dev/null +++ b/.SRCINFO @@ -0,0 +1,17 @@ +pkgbase = catboost-git + pkgdesc = CatBoost is an open-source gradient boosting on decision trees library with categorical features support out of the box. + pkgver = 0.12.2 + pkgrel = 1 + epoch = 0 + url = https://catboost.yandex + arch = i686 + arch = x86_64 + license = Apache-2.0 + depends = clang + depends = python-six + depends = python-numpy + source = catboost::git+https://github.com/catboost/catboost.git + md5sums = SKIP + +pkgname = catboost-git + diff --git a/.gitignore b/.gitignore new file mode 100644 index 000000000000..53a03d3e8659 --- /dev/null +++ b/.gitignore @@ -0,0 +1,4 @@ +catboost-git*.pkg.tar.xz +catboost/ +pkg/ +src/ diff --git a/PKGBUILD b/PKGBUILD new file mode 100644 index 000000000000..cd664b8b916e --- /dev/null +++ b/PKGBUILD @@ -0,0 +1,67 @@ +# Maintainer: Daniel Bershatsky <bepshatsky@yandex.ru> +# Build CLI and Python wrapper without GPU support. +pkgname=catboost-git +pkgver=0.12.2 +pkgrel=1 +epoch=0 +pkgdesc="CatBoost is an open-source gradient boosting on decision trees library with categorical features support out of the box." +arch=('i686' 'x86_64') +url="https://catboost.yandex" +license=('Apache-2.0') +depends=('python-six' 'python-numpy') +makedepends=() +checkdepends=() +optdepends=('python-pandas' 'python-ipywidgets') +provides=() +conflicts=() +replaces=() +backup=() +options=() + +source=('catboost::git+https://github.com/catboost/catboost.git') +md5sums=('SKIP') + +validpgpkeys=() + +pkgver() { + cd "$srcdir/catboost" + printf "$(git describe | sed -E 's/^v([0-9]+.[0-9]+.[0-9]+).*$/\1/')" +} + +build() { + export YA_CACHE_DIR=/tmp/.ya + + echo 'command line utility' + cd "$srcdir/catboost/catboost/app" + ../../ya make -r \ + -DNO_DEBUGINFO \ + -DUSE_ARCADIA_PYTHON=no \ + -DOS_SDK=local \ + -DPYTHON_CONFIG=python-config \ + -DHAVE_CUDA=no + + echo 'python package' + cd "$srcdir/catboost/catboost/python-package/catboost" + ../../../ya make -r \ + -DNO_DEBUGINFO \ + -DUSE_ARCADIA_PYTHON=no \ + -DOS_SDK=local \ + -DPYTHON_CONFIG=python-config \ + -DHAVE_CUDA=no + + cd "$srcdir/catboost/catboost/python-package" + python3 mk_wheel.py +} + +package() { + cd "$srcdir/catboost/catboost/app" + mkdir -p "$pkgdir/usr/bin" + cp catboost "$pkgdir/usr/bin/catboost" + + cd "$srcdir/catboost/catboost/python-package" + pip3 install \ + --ignore-installed \ + --no-deps \ + --prefix $pkgdir/usr \ + catboost-*.whl +} |