# Maintainer: Butui Hu # Contributor: Chih-Hsuan Yen # Contributor: Jean Lucas # Based on python-torchvision-git; original contributors: # Contributor: Stephen Zhang pkgname=python-torchvision _pkgname=vision pkgver=0.4.2 pkgrel=1 pkgdesc='Datasets, transforms, and models specific to computer vision' arch=('x86_64') url='https://github.com/pytorch/vision' license=('BSD') depends=( 'python-av' 'python-numpy' 'python-pillow' 'python-pytorch' 'python-scipy' 'python-six' 'python-tqdm' ) makedepends=( 'python-setuptools' 'qt5-base' ) checkdepends=( 'python-mock' 'python-pytest' 'python-scipy' ) source=("https://github.com/pytorch/vision/archive/v${pkgver}.tar.gz") sha512sums=('3327364bd1c10f4f0fdf9f8c1bed3848cbd887ee335634a0da6eb7a2611037408eee6261cd178d12225be5f2c809a05ff9fb87420b8accd447a799ca83def3ad') build() { cd "${_pkgname}-${pkgver}" python setup.py build } check() { cd "${_pkgname}-${pkgver}" PYTHONPATH="${PWD}/build/lib.linux-${CARCH}-3.7" pytest -v } package() { cd "${_pkgname}-${pkgver}" python setup.py install --root="${pkgdir}" --optimize=1 --skip-build install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}" } # vim:set ts=2 sw=2 et: