mirror of
https://github.com/ilyakooo0/nixpkgs.git
synced 2024-10-14 08:31:40 +03:00
Merge pull request #333635 from NixOS/blackjax
python312Packages.blackjax: 1.2.2 -> 1.2.3
This commit is contained in:
commit
ff84999b7e
@ -1,23 +1,30 @@
|
|||||||
{
|
{
|
||||||
lib,
|
lib,
|
||||||
stdenv,
|
|
||||||
buildPythonPackage,
|
buildPythonPackage,
|
||||||
pythonOlder,
|
pythonOlder,
|
||||||
fetchFromGitHub,
|
fetchFromGitHub,
|
||||||
pytest-xdist,
|
|
||||||
pytestCheckHook,
|
# build-system
|
||||||
setuptools-scm,
|
setuptools-scm,
|
||||||
|
|
||||||
|
# dependencies
|
||||||
fastprogress,
|
fastprogress,
|
||||||
jax,
|
jax,
|
||||||
jaxlib,
|
jaxlib,
|
||||||
jaxopt,
|
jaxopt,
|
||||||
optax,
|
optax,
|
||||||
typing-extensions,
|
typing-extensions,
|
||||||
|
|
||||||
|
# checks
|
||||||
|
pytestCheckHook,
|
||||||
|
pytest-xdist,
|
||||||
|
|
||||||
|
stdenv,
|
||||||
}:
|
}:
|
||||||
|
|
||||||
buildPythonPackage rec {
|
buildPythonPackage rec {
|
||||||
pname = "blackjax";
|
pname = "blackjax";
|
||||||
version = "1.2.2";
|
version = "1.2.3";
|
||||||
pyproject = true;
|
pyproject = true;
|
||||||
|
|
||||||
disabled = pythonOlder "3.9";
|
disabled = pythonOlder "3.9";
|
||||||
@ -26,7 +33,7 @@ buildPythonPackage rec {
|
|||||||
owner = "blackjax-devs";
|
owner = "blackjax-devs";
|
||||||
repo = "blackjax";
|
repo = "blackjax";
|
||||||
rev = "refs/tags/${version}";
|
rev = "refs/tags/${version}";
|
||||||
hash = "sha256-W89L/3bpvecZdCpLCQOppev+fPc+SXd/T+zZLBH2wJs=";
|
hash = "sha256-f1piE79TLVLtIe9/DaLhXss/ifhU719nEylyl70SVJc=";
|
||||||
};
|
};
|
||||||
|
|
||||||
build-system = [ setuptools-scm ];
|
build-system = [ setuptools-scm ];
|
||||||
|
@ -86,13 +86,21 @@ buildPythonPackage rec {
|
|||||||
"test_dtype_consistency"
|
"test_dtype_consistency"
|
||||||
# AssertionError: Array(0.01411963, dtype=float32) not less than or equal to 0.01
|
# AssertionError: Array(0.01411963, dtype=float32) not less than or equal to 0.01
|
||||||
"test_multiclass_logreg6"
|
"test_multiclass_logreg6"
|
||||||
|
]
|
||||||
|
++ lib.optionals stdenv.isDarwin [
|
||||||
|
# Fatal Python error: Aborted
|
||||||
|
"test_dtype_consistency"
|
||||||
|
|
||||||
|
# AssertionError (flaky numerical tests)
|
||||||
|
"test_logreg_with_intercept_manual_loop3"
|
||||||
|
"test_binary_logit_log_likelihood"
|
||||||
];
|
];
|
||||||
|
|
||||||
meta = with lib; {
|
meta = {
|
||||||
homepage = "https://jaxopt.github.io";
|
homepage = "https://jaxopt.github.io";
|
||||||
description = "Hardware accelerated, batchable and differentiable optimizers in JAX";
|
description = "Hardware accelerated, batchable and differentiable optimizers in JAX";
|
||||||
changelog = "https://github.com/google/jaxopt/releases/tag/jaxopt-v${version}";
|
changelog = "https://github.com/google/jaxopt/releases/tag/jaxopt-v${version}";
|
||||||
license = licenses.asl20;
|
license = lib.licenses.asl20;
|
||||||
maintainers = with maintainers; [ bcdarwin ];
|
maintainers = with lib.maintainers; [ bcdarwin ];
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user