Add python-graphviz dependency

2 files changed, 4 insertions, 3 deletions

diff --git a/.SRCINFO b/.SRCINFO
index b398f91ec6b6..a806c142b890 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,7 +1,7 @@
 pkgbase = python-catboost
 	pkgdesc = Fast, scalable, high performance gradient boosting on decision trees
 	pkgver = 0.15
-	pkgrel = 1
+	pkgrel = 2
 	url = https://catboost.ai/
 	arch = x86_64
 	license = Apache
@@ -9,6 +9,7 @@ pkgbase = python-catboost
 	makedepends = python-pip
 	depends = python-numpy
 	depends = python-six
+	depends = python-graphviz
 	optdepends = python-ipywidgets: visualization in Jupyter
 	options = !emptydirs
 	source = https://files.pythonhosted.org/packages/cp37/c/catboost/catboost-0.15-cp37-none-manylinux1_x86_64.whl
diff --git a/PKGBUILD b/PKGBUILD
index 0e1c11dc44ac..be58229a5707 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -3,12 +3,12 @@ pkgname=python-catboost
 _name=${pkgname#python-}
 _py=cp37
 pkgver=0.15
-pkgrel=1
+pkgrel=2
 pkgdesc="Fast, scalable, high performance gradient boosting on decision trees"
 arch=('x86_64')
 url="https://catboost.ai/"
 license=('Apache')
-depends=('python-numpy' 'python-six')
+depends=('python-numpy' 'python-six' 'python-graphviz')
 makedepends=('python-wheel' 'python-pip')
 optdepends=('python-ipywidgets: visualization in Jupyter')
 options=(!emptydirs)