diff options
| author | Otakar Jašek | 2021-08-12 15:19:51 +0200 |
|---|---|---|
| committer | Otakar Jašek | 2021-08-12 15:19:51 +0200 |
| commit | 6ace9e28edb0de7e0cfb384d0898dca3611e035a (patch) | |
| tree | 88c50f21f47b629be56f2540f5e38712914936d5 /PKGBUILD | |
| parent | 9fc63a17fda941fe0601aa3bf4235d395d4c3bac (diff) | |
| download | aur-6ace9e28edb0de7e0cfb384d0898dca3611e035a.tar.gz | |
verbump
Diffstat (limited to 'PKGBUILD')
| -rw-r--r-- | PKGBUILD | 18 |
1 files changed, 12 insertions, 6 deletions
@@ -1,23 +1,29 @@ # Maintainer: otaj +# If you also want to test GPU portion of the package, set this to 1. Make sure you have a capable GPU with large enough memory +TEST_GPU=0 pkgname=python-kornia _name=kornia -pkgver=0.5.0 +pkgver=0.5.8 pkgrel=1 arch=(any) url='https://github.com/kornia/kornia' pkgdesc='Open Source Differentiable Computer Vision Library for PyTorch' license=(Apache) -makedepends=('python-setuptools') -depends=('python-pytorch' 'python-numpy' 'python-torchvision' 'opencv' 'ipython' 'jupyter' 'python-matplotlib') +makedepends=('python-setuptools' 'python-pip') +depends=('python-pytorch') checkdepends=('python-pytest') options=(!emptydirs) -source=(${_name}-${pkgver}.tar.gz::"${url}/archive/v${pkgver}.tar.gz") -sha256sums=('c6c428790becb28c9af86bf9284100ea485f6c33c735d5aa2c9754bbd5395e22') +# because kornia does not ship `pytest.ini` with their release, we have to get a fat copy of full repo +source=(${_name}-${pkgver}.tar.gz::"${url}/archive/refs/tags/${pkgver}.tar.gz") +sha256sums=('e0a2686c7b2856b4ccc2883817d96c7d888c6df4745beedcfe307cbfe0b50cca') check() { cd "${srcdir}/${_name}-${pkgver}" - pytest -v --device all --dtype float32,float64 test/ + pytest -v --device cpu --dtype float32,float64 test/ + if ! [ "$TEST_GPU" -eq "0" ] ; then + pytest -v --device cuda --dtype float32,float64 test/ + fi } package() { |