Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.2 kB · 69 lines
Nix
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970{ lib, buildPythonPackage, fetchFromGitHub,
# build-system hatchling,
# dependencies chex, equinox, jax, jaxtyping, optax, scikit-learn, tqdm,
# tests pytestCheckHook,}:
buildPythonPackage rec { pname = "flowmc"; version = "0.4.5"; pyproject = true;
src = fetchFromGitHub { owner = "kazewong"; repo = "flowMC"; tag = "flowMC-${version}"; hash = "sha256-D3K9cvmUvwsVAjvXdSDgYlqrzTYXVlSVQbfx7TANz8A="; };
build-system = [ hatchling ];
pythonRelaxDeps = [ "jax" ];
pythonRemoveDeps = [ # Not actual runtime dependencies "pre-commit" "pyright" "pytest" "ruff" ];
dependencies = [ chex equinox jax jaxtyping optax scikit-learn tqdm ];
pythonImportsCheck = [ "flowMC" ];
nativeCheckInputs = [ pytestCheckHook ];
meta = { description = "Normalizing-flow enhanced sampling package for probabilistic inference in Jax"; homepage = "https://github.com/kazewong/flowMC"; changelog = "https://github.com/kazewong/flowMC/releases/tag/flowMC-${version}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ GaetanLepage ]; };}