summarylogtreecommitdiffstats
path: root/PKGBUILD
diff options
context:
space:
mode:
Diffstat (limited to 'PKGBUILD')
-rw-r--r--PKGBUILD56
1 files changed, 37 insertions, 19 deletions
diff --git a/PKGBUILD b/PKGBUILD
index 2334ea0d25e3..180f536bacc2 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,29 +1,47 @@
-# Maintainer: Filip Graliński <filipg@amu.edu.pl>
+# Maintainer: Luis Martinez <luis dot martinez at disroot dot org>
+# Contributor: Filip Graliński <filipg@amu.edu.pl>
pkgname=fairseq
-pkgver=0.10.2
+pkgver=0.12.2
pkgrel=1
-pkgdesc="sequence modeling toolkit"
-arch=('i686' 'x86_64')
-url="https://github.com/pytorch/fairseq"
+pkgdesc="Sequence modeling toolkit"
+arch=('x86_64')
+url="https://github.com/facebookresearch/fairseq"
license=('MIT')
-depends=('cuda'
- 'nccl'
- 'python-antlr4'
- 'python-bitarray'
- 'python-hydra'
- 'python-omegaconf'
- 'python-pytorch')
-optdepends=('python-apex: Faster training')
-source=("https://github.com/pytorch/$pkgname/archive/refs/tags/v${pkgver}.tar.gz")
-sha256sums=('4328a7cb5fe6aea874817a863eb7f1ace91d621bc504345e6d022ce64e065da0')
+depends=(
+ 'python>=3.6'
+ 'python-bitarray'
+ 'python-cffi'
+ 'python-hydra-core'
+ 'python-numpy'
+ 'python-omegaconf'
+ 'python-pytorch'
+ 'python-regex'
+ 'python-sacrebleu'
+ 'python-tqdm'
+ 'python-torchaudio')
+optdepends=(
+ 'cuda: Model training'
+ 'nccl: Model training'
+ 'python-apex: Faster training')
+makedepends=(
+ 'cython'
+ 'python-build'
+ 'python-installer'
+ 'python-setuptools'
+ 'python-wheel')
+source=("$pkgname-$pkgver.tar.gz::$url/archive/v$pkgver.tar.gz")
+sha256sums=('b5bb076f8b0d5b06a897b11627da278816752b2202650c34becd8f4240f86961')
build() {
- cd $pkgname-$pkgver
- python setup.py build
+ cd "$pkgname-$pkgver"
+ python -m build --wheel --no-isolation
}
package() {
- cd $pkgname-$pkgver
- python setup.py install --root="$pkgdir" --optimize=1
+ cd "$pkgname-$pkgver"
+ PYTHONHASHSEED=0 python -m installer --destdir="$pkgdir" dist/*.whl
+ local _site="$(python -c 'import site; print(site.getsitepackages()[0])')"
+ install -d "$pkgdir/usr/share/licenses/$pkgname/"
+ ln -s "$_site/$pkgname-$pkgver.dist-info/LICENSE" "$pkgdir/usr/share/licenses/$pkgname/"
}