Merge pull request #309751 from GaetanLepage/blackjax

python311Packages.blackjax: 1.1.1 -> 1.2.0
This commit is contained in:
Pol Dellaiera 2024-05-07 09:27:31 +02:00 committed by GitHub
commit fa4bb2cdd2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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"