2 files changed, 5 insertions, 2 deletions
diff --git a/.SRCINFO b/.SRCINFO
index de5a9a1ae55e..1fefb1d50839 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,10 +1,11 @@
 pkgbase = python-pytorch-thop-git
 	pkgdesc = PyTorch MACs counter
 	pkgver = r112.d1920d3
-	pkgrel = 1
+	pkgrel = 2
 	url = https://github.com/Lyken17/pytorch-OpCounter
 	arch = any
 	license = MIT
+	makedepends = python-setuptools
 	depends = python-pytorch
 	source = git+https://github.com/Lyken17/pytorch-OpCounter.git
 	sha256sums = SKIP
diff --git a/PKGBUILD b/PKGBUILD
index 35580834d1b8..7a0de2d2c4e2 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -4,12 +4,13 @@ pkgname=python-pytorch-thop-git
 _reponame="pytorch-OpCounter"
 _modulename="thop"
 pkgver=r112.d1920d3
-pkgrel=1
+pkgrel=2
 pkgdesc="PyTorch MACs counter"
 url="https://github.com/Lyken17/${_reponame}"
 license=('MIT')
 arch=('any')
 depends=('python-pytorch')
+makedepends=('python-setuptools')
 source=("git+${url}.git")
 sha256sums=('SKIP')
 
@@ -28,6 +29,7 @@ build() {
 
 package() {
   cd ${srcdir}/${_reponame}
+  install -Dm644 LICENSE "$pkgdir/usr/share/licenses/$pkgname/LICENSE"
   python setup.py install --root="$pkgdir/" --optimize=1 --skip-build
 }