Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.6 kB · 75 lines
Nix
at llvm-cleanup
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576{ lib, stdenv, buildPythonPackage, pythonOlder, fetchFromGitHub, pytest-xdist, pytestCheckHook, setuptools-scm, fastprogress, jax, jaxlib, jaxopt, optax, typing-extensions,}:
buildPythonPackage rec { pname = "blackjax"; version = "1.2.1"; pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub { owner = "blackjax-devs"; repo = "blackjax"; rev = "refs/tags/${version}"; hash = "sha256-VoWBCjFMyE5LVJyf7du/pKlnvDHj22lguiP6ZUzH9ak="; };
build-system = [ setuptools-scm ];
dependencies = [ fastprogress jax jaxlib jaxopt optax typing-extensions ];
nativeCheckInputs = [ pytestCheckHook pytest-xdist ];
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" ] ++ lib.optionals (stdenv.isLinux && stdenv.isAarch64) [ # Numerical test (AssertionError) # https://github.com/blackjax-devs/blackjax/issues/668 "test_chees_adaptation" ];
pythonImportsCheck = [ "blackjax" ];
meta = with lib; { homepage = "https://blackjax-devs.github.io/blackjax"; description = "Sampling library designed for ease of use, speed and modularity"; changelog = "https://github.com/blackjax-devs/blackjax/releases/tag/${version}"; license = licenses.asl20; maintainers = with maintainers; [ bcdarwin ]; };}