Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.1 kB · 96 lines
Nix
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697{ lib, buildPythonPackage, fetchFromGitHub,
# build-system setuptools-scm,
# dependencies fastprogress, jax, jaxlib, jaxopt, optax, typing-extensions,
# checks pytestCheckHook, pytest-xdist,
stdenv,}:
buildPythonPackage rec { pname = "blackjax"; version = "1.2.5"; pyproject = true;
src = fetchFromGitHub { owner = "blackjax-devs"; repo = "blackjax"; tag = version; hash = "sha256-2GTjKjLIWFaluTjdWdUF9Iim973y81xv715xspghRZI="; };
build-system = [ setuptools-scm ];
dependencies = [ fastprogress jax jaxlib jaxopt optax typing-extensions ];
nativeCheckInputs = [ pytestCheckHook pytest-xdist ];
pytestFlags = [ # DeprecationWarning: JAXopt is no longer maintained "-Wignore::DeprecationWarning" ];
disabledTestPaths = [ "tests/test_benchmarks.py"
# Assertion errors on numerical values "tests/mcmc/test_integrators.py" ];
disabledTests = [ # too slow "test_adaptive_tempered_smc"
# AssertionError on numerical values "test_barker" "test_mclmc" "test_mcse4" "test_normal_univariate" "test_nuts__with_device" "test_nuts__with_jit" "test_nuts__without_device" "test_nuts__without_jit" "test_smc_waste_free__with_jit"
# Numerical test (AssertionError) # First report, when the failure was only happening on aarch64-linux: # https://github.com/blackjax-devs/blackjax/issues/668 # Second report, when the test started happening on x86_64-linux too after Jax was updated to 0.7.0 # https://github.com/blackjax-devs/blackjax/issues/795 "test_chees_adaptation" ];
pythonImportsCheck = [ "blackjax" ];
meta = { 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 = lib.licenses.asl20; maintainers = with lib.maintainers; [ bcdarwin ]; };}