Initial commit

1 files changed, 67 insertions, 0 deletions

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
+}