2 files changed, 5 insertions, 2 deletions
diff --git a/.SRCINFO b/.SRCINFO
index 1a304b7e1ecd..7f22ce66f3b7 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,10 +1,11 @@
 pkgbase = python-ptflops
 	pkgdesc = Flops counter for convolutional networks in pytorch framework
 	pkgver = 0.6.2
-	pkgrel = 1
+	pkgrel = 2
 	url = https://github.com/sovrasov/flops-counter.pytorch
 	arch = any
 	license = MIT
+	makedepends = python-setuptools
 	depends = python-pytorch
 	source = https://github.com/sovrasov/flops-counter.pytorch/archive/v0.6.2.tar.gz
 	sha256sums = 5abbef678601a4cb7bead99fbbec0265a2a1746718eecf76975d888f0e2d048b
diff --git a/PKGBUILD b/PKGBUILD
index 2a122f841f2a..028927fba274 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -4,12 +4,13 @@ pkgname=python-ptflops
 _reponame="flops-counter.pytorch"
 _modulename="ptflops"
 pkgver=0.6.2
-pkgrel=1
+pkgrel=2
 pkgdesc="Flops counter for convolutional networks in pytorch framework"
 url="https://github.com/sovrasov/${_reponame}"
 license=('MIT')
 arch=('any')
 depends=('python-pytorch')
+makedepends=('python-setuptools')
 source=("${url}/archive/v${pkgver}.tar.gz")
 sha256sums=('5abbef678601a4cb7bead99fbbec0265a2a1746718eecf76975d888f0e2d048b')
 
@@ -20,6 +21,7 @@ build() {
 
 package() {
   cd ${srcdir}/${_reponame}-${pkgver}
+  install -Dm644 LICENSE "$pkgdir/usr/share/licenses/$pkgname/LICENSE"
   python setup.py install --root="$pkgdir/" --optimize=1 --skip-build
 }