sklearn

2 files changed, 3 insertions, 1 deletions

diff --git a/.SRCINFO b/.SRCINFO
index dec1cf34baf9..c4632a48b099 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -12,6 +12,8 @@ pkgbase = agrum
 	depends = ipython
 	depends = python-ipykernel
 	depends = python-pandas
+	depends = python-scikit-learn
+	depends = python-cairosvg
 	depends = unixodbc
 	source = https://gitlab.com/agrumery/aGrUM/-/archive/0.22.6/aGrUM-0.22.6.tar.bz2
 	sha256sums = be15f0e47143a11f11b5222a56f3270c922069287a63ebae73c9b655ec5bbf7d
diff --git a/PKGBUILD b/PKGBUILD
index de48872808c7..3195ba39236a 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -5,7 +5,7 @@ pkgdesc="C++ Bayesian networks library"
 license=('LGPL')
 arch=('x86_64')
 url="http://agrum.gitlab.io/"
-depends=('python-pydot' 'python-matplotlib' 'python-six' 'ipython' 'python-ipykernel' 'python-pandas' 'unixodbc')
+depends=('python-pydot' 'python-matplotlib' 'python-six' 'ipython' 'python-ipykernel' 'python-pandas' 'python-scikit-learn' 'python-cairosvg' 'unixodbc')
 makedepends=('cmake')
 source=("https://gitlab.com/agrumery/aGrUM/-/archive/${pkgver}/aGrUM-${pkgver}.tar.bz2")
 sha256sums=('be15f0e47143a11f11b5222a56f3270c922069287a63ebae73c9b655ec5bbf7d')