cleaning, test only common dtypes

2 files changed, 8 insertions, 11 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 5ee6201bd358..1ac8e11c83c5 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,19 +1,16 @@
 pkgbase = python-kornia-git
 	pkgdesc = Open Source Differentiable Computer Vision Library for PyTorch
-	pkgver = 0.2.1.r1.g9797eca
+	pkgver = v0.3.1.r10.gf4f70fe
 	pkgrel = 1
 	url = https://github.com/kornia/kornia
 	arch = any
 	license = Apache
 	checkdepends = python-pytest
+	checkdepends = python-pytest-runner
 	makedepends = git
 	makedepends = python-setuptools
 	depends = python-pytorch
-	depends = python-torchvision
-	depends = opencv
-	depends = python-matplotlib
-	depends = ipython
-	depends = jupyter
+	depends = python-numpy
 	provides = python-kornia
 	conflicts = python-kornia
 	options = !emptydirs
diff --git a/PKGBUILD b/PKGBUILD
index 308c98c4b3ab..078b7e44819f 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,15 +2,15 @@
 
 pkgname=python-kornia-git
 _name=kornia
-pkgver=0.2.1.r1.g9797eca
+pkgver=v0.3.1.r10.gf4f70fe
 pkgrel=1
 arch=(any)
 url='https://github.com/kornia/kornia'
 pkgdesc='Open Source Differentiable Computer Vision Library for PyTorch'
 license=(Apache)
 makedepends=('git' 'python-setuptools')
-depends=('python-pytorch' 'python-torchvision' 'opencv' 'python-matplotlib' 'ipython' 'jupyter')
-checkdepends=('python-pytest')
+depends=('python-pytorch' 'python-numpy')
+checkdepends=('python-pytest' 'python-pytest-runner')
 provides=('python-kornia')
 conflicts=('python-kornia')
 options=(!emptydirs)
@@ -23,8 +23,8 @@ pkgver() {
 }
 
 check() {
-  cd "${srcdir}/${_name}-${pkgver}"
-  pytest
+  cd "${srcdir}/${_name}"
+  pytest -v --device all --dtype float32,float64 test/
 }
 
 package() {