fix test deps

calad0i · Nov 19, 2024 · 06f8b33 · 06f8b33
1 parent 3a8fb60
commit 06f8b33
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -37,7 +37,15 @@ write_to = "src/quantizers/_version.py"
 dependencies = ["numpy"]
 
 grad = ['keras>=3']
-test = ['pytest', 'pytest-cov', 'pytest-xdist', 'keras>=3']
+test = [
+  'pytest',
+  'pytest-cov',
+  'pytest-xdist',
+  'pytest-env',
+  'keras>=3.0',
+  'cppyy',
+  'jax',
+]
 
 [tool.pytest.ini_options]
 filterwarnings = [