summarylogtreecommitdiffstats
path: root/PKGBUILD
blob: 9b14f7a357a88f11d423d1eb83a89ed3e535071d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# Maintainer: Alexander F. Rødseth <xyproto@archlinux.org>

pkgname=python-ludwig
pkgver=0.1.0
pkgrel=5
pkgdesc='Train and test deep learning models'
arch=(any)
url='https://github.com/uber/ludwig'
license=(MIT)
depends=(cython python-cloudpickle python-dask python-h5py python-matplotlib python-numpy python-pandas python-scipy python-scikit-learn python-tqdm python-tabulate python-tensorflow python-yaml python-pytest python-scikit-image python-seaborn python-spacy-en_core_web_sm 'python-spacy>=2.0')
depends+=(python2-numba) # python2-numba is requored for the check() function in python-sparse to pass
makedepends=(git)
optdepends=('ludwig-example: example use of Ludwig')
source=("git+https://github.com/uber/ludwig#tag=v$pkgver")
sha256sums=('SKIP')

build() {
  cd ludwig
  python setup.py build
}

package() {
  cd ludwig

  python setup.py install --root="$pkgdir" --optimize=1
  install -Dm644 LICENSE "$pkgdir/usr/share/licenses/$pkgname/LICENSE"

  # fix issue where spacy >= 2.0 is not detected
  find "$pkgdir" -name requires.txt -type f -exec sed -i '/spacy/d' {} \;
}

# vim: ts=2 sw=2 et: