Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.4 kB · 84 lines
Nix
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485{ lib, buildPythonPackage, pythonOlder, fetchFromGitHub,
# build-system setuptools,
# dependencies chex, dotmap, flax, jax, jaxlib, matplotlib, numpy, pyyaml,
# checks # brax, (unpackaged) # gymnax, (unpackaged) pytestCheckHook, torch, torchvision,}:
buildPythonPackage rec { pname = "evosax"; version = "0.1.6"; pyproject = true;
disabled = pythonOlder "3.7";
src = fetchFromGitHub { owner = "RobertTLange"; repo = "evosax"; rev = "refs/tags/v.${version}"; hash = "sha256-v8wRiWZlJPF9pIXocQ6/caHl1W4QBNjkmuImJ6MAueo="; };
build-system = [ setuptools ];
dependencies = [ chex dotmap flax jax jaxlib matplotlib numpy pyyaml ];
pythonImportsCheck = [ "evosax" ];
preCheck = '' export HOME=$(mktemp -d) '';
nativeCheckInputs = [ # brax # gymnax pytestCheckHook torch torchvision ];
disabledTests = [ # Requires unpackaged gymnax "test_env_ffw_rollout"
# Tries to download a data set from the internet "test_vision_fitness" ];
meta = { description = "Evolution Strategies in JAX"; homepage = "https://github.com/RobertTLange/evosax"; changelog = "https://github.com/RobertTLange/evosax/blob/${src.rev}/CHANGELOG.md"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ GaetanLepage ]; };}