summarylogtreecommitdiffstats
path: root/PKGBUILD
diff options
context:
space:
mode:
Diffstat (limited to 'PKGBUILD')
-rw-r--r--PKGBUILD19
1 files changed, 10 insertions, 9 deletions
diff --git a/PKGBUILD b/PKGBUILD
index 000b1c5dd11..fd672f2fad1 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -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}"
}