Gaetan Lepage 2024-05-06 20:57:46 +02:00
parent b771106b51
commit fd36966960

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"