2 files changed, 4 insertions, 2 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 907b1ddc3884..4dd4aa1ffd82 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,10 +1,11 @@
 pkgbase = python-tensorboardx
 	pkgdesc = Tensorboard for PyTorch
 	pkgver = 1.4
-	pkgrel = 2
+	pkgrel = 3
 	url = https://github.com/lanpa/tensorboardX
 	arch = any
 	license = MIT
+	makedepends = python-setuptools
 	depends = python-pytorch
 	depends = python-torchvision
 	depends = tensorboard
diff --git a/PKGBUILD b/PKGBUILD
index d9ca025da0c4..c739401c2243 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,12 +2,13 @@
 _name=tensorboardX
 pkgname=python-tensorboardx
 pkgver=1.4
-pkgrel=2
+pkgrel=3
 pkgdesc="Tensorboard for PyTorch"
 arch=('any')
 url="https://github.com/lanpa/tensorboardX"
 license=('MIT')
 depends=('python-pytorch' 'python-torchvision' 'tensorboard')
+makedepends=('python-setuptools')
 source+=("https://files.pythonhosted.org/packages/source/${_name::1}/${_name}/${_name}-${pkgver}.tar.gz")
 sha256sums=('bacf2825c8fb9c40bea22c50796fc4afe5bf11ff5f5c3b7ad7c933ad034b6fae')