Move torchvision to optional dependency.

2 files changed, 5 insertions, 2 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 99eec13b313a..1452814209ef 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -8,12 +8,12 @@ pkgbase = python-minkowskiengine
 	makedepends = python-setuptools
 	depends = python
 	depends = python-pytorch-cuda
-	depends = python-torchvision
 	depends = python-numpy
 	depends = cuda
 	depends = cudnn
 	depends = cblas
 	depends = intel-tbb
+	optdepends = python-torchvision: sample code dependency
 	source = https://github.com/StanfordVL/MinkowskiEngine/archive/v0.3.0.tar.gz
 	sha512sums = 7a5b1cfae709937b86c844d8ce6b340ff3fc2fc044d495099f16b4e592180da071122ed5fb87a662dbb78ef1c7af9887312a66a77630e8b1402f96770668073b
 
diff --git a/PKGBUILD b/PKGBUILD
index e95edcd6b14b..1d4f42dd8b21 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -6,7 +6,10 @@ pkgver=0.3.0
 pkgrel=1
 pkgdesc="Auto-diff library for generalized sparse convolutions"
 url="https://github.com/StanfordVL/MinkowskiEngine"
-depends=('python' 'python-pytorch-cuda' 'python-torchvision' 'python-numpy' 'cuda' 'cudnn' 'cblas' 'intel-tbb')
+depends=('python' 'python-pytorch-cuda' 'python-numpy' 'cuda' 'cudnn' 'cblas' 'intel-tbb')
+optdepends=(
+    'python-torchvision: Sample code dependency'
+)
 makedepends=('python-setuptools')
 license=('MIT')
 arch=('x86_64')