add dep

2 files changed, 2 insertions, 1 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 44213b70befa..cdcd46dc9787 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -10,6 +10,7 @@ pkgbase = python-gpustat
 	depends = python-six
 	depends = nvidia
 	depends = python-blessings
+	depends = python-nvidia-ml-py3-git
 	source = https://github.com/wookayin/gpustat/archive/v0.5.0.tar.gz
 	sha256sums = 1b6cac5f3584028504f3cb8738798dfa4989f74084617ae8fce509c5b426abc7
 
diff --git a/PKGBUILD b/PKGBUILD
index abe8ca914991..7053b10bb09b 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -8,7 +8,7 @@ pkgdesc="A simple command-line utility for querying and monitoring GPU status"
 arch=('any')
 url=https://github.com/wookayin/gpustat
 license=('MIT')
-depends=('python' 'python-six' 'nvidia' 'python-blessings')
+depends=('python' 'python-six' 'nvidia' 'python-blessings' 'python-nvidia-ml-py3-git')
 makedepends=('python-setuptools')
 source=("https://github.com/wookayin/${_pkgname}/archive/v${pkgver}.tar.gz")
 sha256sums=('1b6cac5f3584028504f3cb8738798dfa4989f74084617ae8fce509c5b426abc7')