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

View file

@ -1,4 +1,5 @@
{ lib { lib
, stdenv
, buildPythonPackage , buildPythonPackage
, pythonOlder , pythonOlder
, fetchFromGitHub , fetchFromGitHub
@ -15,7 +16,7 @@
buildPythonPackage rec { buildPythonPackage rec {
pname = "blackjax"; pname = "blackjax";
version = "1.1.1"; version = "1.2.0";
pyproject = true; pyproject = true;
disabled = pythonOlder "3.9"; disabled = pythonOlder "3.9";
@ -24,12 +25,14 @@ buildPythonPackage rec {
owner = "blackjax-devs"; owner = "blackjax-devs";
repo = "blackjax"; repo = "blackjax";
rev = "refs/tags/${version}"; rev = "refs/tags/${version}";
hash = "sha256-6+ElY1F8oRCtWT4a/LIG6hYMthlq5mDx2baKAc6zIns="; hash = "sha256-vXyxK3xALKG61YGK7fmoqQNGfOiagHFrvnU02WKZThw=";
}; };
nativeBuildInputs = [ setuptools-scm ]; build-system = [
setuptools-scm
];
propagatedBuildInputs = [ dependencies = [
fastprogress fastprogress
jax jax
jaxlib jaxlib
@ -42,7 +45,14 @@ buildPythonPackage rec {
pytestCheckHook pytestCheckHook
pytest-xdist 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 = [ disabledTests = [
# too slow # too slow
"test_adaptive_tempered_smc" "test_adaptive_tempered_smc"