diff options
Diffstat (limited to 'pkgs/development/python-modules/blackjax/default.nix')
-rw-r--r-- | pkgs/development/python-modules/blackjax/default.nix | 20 |
1 files changed, 15 insertions, 5 deletions
diff --git a/pkgs/development/python-modules/blackjax/default.nix b/pkgs/development/python-modules/blackjax/default.nix index 7ecc4eda95198..6f3403d3c143f 100644 --- a/pkgs/development/python-modules/blackjax/default.nix +++ b/pkgs/development/python-modules/blackjax/default.nix @@ -1,4 +1,5 @@ { lib +, stdenv , buildPythonPackage , pythonOlder , fetchFromGitHub @@ -15,7 +16,7 @@ buildPythonPackage rec { pname = "blackjax"; - version = "1.1.1"; + version = "1.2.0"; pyproject = true; disabled = pythonOlder "3.9"; @@ -24,12 +25,14 @@ buildPythonPackage rec { owner = "blackjax-devs"; repo = "blackjax"; rev = "refs/tags/${version}"; - hash = "sha256-6+ElY1F8oRCtWT4a/LIG6hYMthlq5mDx2baKAc6zIns="; + hash = "sha256-vXyxK3xALKG61YGK7fmoqQNGfOiagHFrvnU02WKZThw="; }; - nativeBuildInputs = [ setuptools-scm ]; + build-system = [ + setuptools-scm + ]; - propagatedBuildInputs = [ + dependencies = [ fastprogress jax jaxlib @@ -42,7 +45,14 @@ buildPythonPackage rec { pytestCheckHook pytest-xdist ]; - disabledTestPaths = [ "tests/test_benchmarks.py" ]; + + disabledTestPaths = [ + "tests/test_benchmarks.py" + ] ++ lib.optionals (stdenv.isLinux && stdenv.isAarch64) [ + # Assertion errors on numerical values + "tests/mcmc/test_integrators.py" + ]; + disabledTests = [ # too slow "test_adaptive_tempered_smc" |