added sympy

2 files changed, 2 insertions, 1 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 3aa3f8c86c9a..b33b28a90126 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -11,6 +11,7 @@ pkgbase = python-apex-git
 	depends = python
 	depends = cuda
 	depends = python-pytorch-cuda
+	depends = python-sympy
 	provides = python-apex
 	conflicts = python-apex
 	source = git+https://github.com/NVIDIA/apex
diff --git a/PKGBUILD b/PKGBUILD
index 06563eea5ab2..5bc97a885eec 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -9,7 +9,7 @@ arch=('x86_64')
 url="https://github.com/NVIDIA/apex"
 _github='NVIDIA/apex'
 license=('BSD')
-depends=('python' 'cuda' 'python-pytorch-cuda')
+depends=('python' 'cuda' 'python-pytorch-cuda' 'python-sympy')
 makedepends=('git' 'python-setuptools' 'python-pip')
 provides=('python-apex')
 conflicts=('python-apex')