fix dependencies

2 files changed, 7 insertions, 4 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 4dd4aa1ffd82..b16af5f01199 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,14 +1,16 @@
 pkgbase = python-tensorboardx
 	pkgdesc = Tensorboard for PyTorch
 	pkgver = 1.4
-	pkgrel = 3
+	pkgrel = 4
 	url = https://github.com/lanpa/tensorboardX
 	arch = any
 	license = MIT
 	makedepends = python-setuptools
 	depends = python-pytorch
 	depends = python-torchvision
-	depends = tensorboard
+	depends = python-protobuf
+	depends = python-six
+	optdepends = tensorboard: to visualise TensorBoard data
 	source = https://files.pythonhosted.org/packages/source/t/tensorboardX/tensorboardX-1.4.tar.gz
 	sha256sums = bacf2825c8fb9c40bea22c50796fc4afe5bf11ff5f5c3b7ad7c933ad034b6fae
 
diff --git a/PKGBUILD b/PKGBUILD
index c739401c2243..1769b472a8a2 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,12 +2,13 @@
 _name=tensorboardX
 pkgname=python-tensorboardx
 pkgver=1.4
-pkgrel=3
+pkgrel=4
 pkgdesc="Tensorboard for PyTorch"
 arch=('any')
 url="https://github.com/lanpa/tensorboardX"
 license=('MIT')
-depends=('python-pytorch' 'python-torchvision' 'tensorboard')
+depends=('python-pytorch' 'python-torchvision' 'python-protobuf' 'python-six')
+optdepends=('tensorboard: to visualise TensorBoard data')
 makedepends=('python-setuptools')
 source+=("https://files.pythonhosted.org/packages/source/${_name::1}/${_name}/${_name}-${pkgver}.tar.gz")
 sha256sums=('bacf2825c8fb9c40bea22c50796fc4afe5bf11ff5f5c3b7ad7c933ad034b6fae')