diff options
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 19 |
1 files changed, 10 insertions, 9 deletions
@@ -5,30 +5,31 @@ # Contributor: Stephen Zhang <zsrkmyn at gmail dot com> pkgname=python-torchvision -pkgver=0.2.2 -pkgrel=6 +_pkgname=vision +pkgver=0.3.0 +pkgrel=2 pkgdesc='Datasets, transforms, and models specific to computer vision' -arch=(any) +arch=('x86_64') url=https://github.com/pytorch/vision license=(BSD) depends=(python-numpy python-pillow python-pytorch python-scipy python-six python-tqdm) makedepends=(python-setuptools) checkdepends=(python-pytest python-scipy) -source=("torchvision-${pkgver}.tar.gz"::"https://github.com/pytorch/vision/archive/v${pkgver}.tar.gz") -sha512sums=('2831391b71257ae4daacd4e7d3f933a910fad548df8df1a70d2358947244de59a70ceb76c544ed3eb1bd07b9e170640234f49b4743d9bb9da0e2b33861c36849') +source=("https://github.com/pytorch/vision/archive/v${pkgver}.tar.gz") +sha512sums=('a402221827319c9792e7ff8edb8218737e09b8dade186a3b7adeb7501f2f60d030e0cc22e5b78163368056370daea4682da0748c72a736df0f4ee4443a8150fe') build() { - cd "vision-${pkgver}" + cd "${_pkgname}-${pkgver}" python setup.py build } check() { - cd "vision-${pkgver}" - PYTHONPATH=. pytest -v test + cd "${_pkgname}-${pkgver}" + PYTHONPATH="${PWD}/build/lib.linux-${CARCH}-3.7" pytest -v } package() { - cd "vision-${pkgver}" + cd "${_pkgname}-${pkgver}" python setup.py install --root="${pkgdir}" --optimize=1 --skip-build install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}" } |