Add missing dependency

2 files changed, 4 insertions, 2 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 0597b42f1470..1736bbec7dfe 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,7 +1,7 @@
 pkgbase = python-jax
 	pkgdesc = Differentiate, compile, and transform Numpy code.
 	pkgver = 0.4.7
-	pkgrel = 1
+	pkgrel = 2
 	url = https://github.com/google/jax/
 	arch = any
 	groups = jax
@@ -12,6 +12,7 @@ pkgbase = python-jax
 	makedepends = python-wheel
 	depends = python-absl
 	depends = python-jaxlib
+	depends = python-ml-dtypes
 	depends = python-numpy>=1.20
 	depends = python-opt_einsum
 	depends = python-scipy>=1.5
diff --git a/PKGBUILD b/PKGBUILD
index 107f47bca7d1..48460612c110 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,7 +2,7 @@
 
 pkgname='python-jax'
 pkgver=0.4.7
-pkgrel=1
+pkgrel=2
 pkgdesc='Differentiate, compile, and transform Numpy code.'
 arch=('any')
 url='https://github.com/google/jax/'
@@ -10,6 +10,7 @@ license=('Apache')
 groups=('jax')
 depends=('python-absl'
          'python-jaxlib'
+         'python-ml-dtypes'
          'python-numpy>=1.20'
          'python-opt_einsum'
          'python-scipy>=1.5')