Add python-yaml dependency and cudnn optional dependency

2 files changed, 11 insertions, 7 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 722d9305565e..08703fc87456 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,9 +1,9 @@
 # Generated by mksrcinfo v8
-# Tue Mar 14 22:28:23 UTC 2017
+# Thu Mar 16 10:21:31 UTC 2017
 pkgbase = python-keras
 	pkgdesc = Deep Learning library (convnets, recurrent neural networks, and more)
 	pkgver = 2.0.0
-	pkgrel = 1
+	pkgrel = 2
 	url = https://github.com/fchollet/keras
 	arch = any
 	license = MIT
@@ -25,13 +25,17 @@ pkgname = python-keras
 	depends = python-numpy
 	depends = python-scipy
 	depends = python-h5py
+	depends = python-yaml
 	optdepends = python-theano
 	optdepends = python-tensorflow
+	optdepends = cudnn
 
 pkgname = python2-keras
 	depends = python2
 	depends = python2-numpy
 	depends = python2-scipy
 	depends = python2-h5py
+	depends = python2-yaml
 	optdepends = python2-theano
+	optdepends = cudnn
 
diff --git a/PKGBUILD b/PKGBUILD
index 86606046b67c..efc61cbf2ff8 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -5,7 +5,7 @@ pkgbase="python-keras"
 pkgname=("python-keras" "python2-keras")
 _pkgname="keras"
 pkgver="2.0.0"
-pkgrel="1"
+pkgrel="2"
 pkgdesc="Deep Learning library (convnets, recurrent neural networks, and more)"
 arch=('any')
 url="https://github.com/fchollet/keras"
@@ -37,16 +37,16 @@ build() {
 }
 
 package_python2-keras() {
-  depends=('python2' 'python2-numpy' 'python2-scipy' 'python2-h5py')
-  optdepends=('python2-theano')
+  depends=('python2' 'python2-numpy' 'python2-scipy' 'python2-h5py' 'python2-yaml')
+  optdepends=('python2-theano' 'cudnn')
   cd "$srcdir/${_pkgname}-${pkgver}-py2"
   python2 setup.py install --root="$pkgdir"/ --optimize=1
   install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE.txt"
 }
 
 package_python-keras() {
-  depends=('python' 'python-numpy' 'python-scipy' 'python-h5py')
-  optdepends=('python-theano' 'python-tensorflow')
+  depends=('python' 'python-numpy' 'python-scipy' 'python-h5py' 'python-yaml')
+  optdepends=('python-theano' 'python-tensorflow' 'cudnn')
   cd "$srcdir/${_pkgname}-${pkgver}"
   python setup.py install --root="$pkgdir"/ --optimize=1
   install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE.txt"