Unverified Commit fa4bb2cd authored by Pol Dellaiera's avatar Pol Dellaiera Committed by GitHub
Browse files

Merge pull request #309751 from GaetanLepage/blackjax

python311Packages.blackjax: 1.1.1 -> 1.2.0
parents 8a70e154 fd369669
Loading
Loading
Loading
Loading
+15 −5
Original line number Diff line number Diff line
{ 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"