From ce75ef61f9686bf14580f23126ec5be614676181 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Wed, 2 Apr 2025 00:46:17 +0200
Subject: [PATCH] python312Packages.evosax: 0.1.6 -> 0.2.0

Diff: https://github.com/RobertTLange/evosax/compare/refs/tags/v.0.1.6...v.0.2.0

Changelog: https://github.com/RobertTLange/evosax/releases/tag/v.0.2.0
---
 .../python-modules/evosax/default.nix            | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/pkgs/development/python-modules/evosax/default.nix b/pkgs/development/python-modules/evosax/default.nix
index 143ece34b4ab..c11d37113080 100644
--- a/pkgs/development/python-modules/evosax/default.nix
+++ b/pkgs/development/python-modules/evosax/default.nix
@@ -27,27 +27,24 @@
 
 buildPythonPackage rec {
   pname = "evosax";
-  version = "0.1.6";
+  version = "0.2.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "RobertTLange";
     repo = "evosax";
     tag = "v.${version}";
-    hash = "sha256-v8wRiWZlJPF9pIXocQ6/caHl1W4QBNjkmuImJ6MAueo=";
+    hash = "sha256-ye5IHM8Pn/+BXI9kcB3W281Gna9hXV8DwsaJ9Xu06fU=";
   };
 
   build-system = [ setuptools ];
 
   dependencies = [
-    chex
     dotmap
     flax
     jax
-    jaxlib
     matplotlib
     numpy
-    pyyaml
   ];
 
   pythonImportsCheck = [ "evosax" ];
@@ -66,6 +63,15 @@ buildPythonPackage rec {
     "test_env_ffw_rollout"
 
     # Tries to download a data set from the internet
+    "test_brax_problem_eval"
+    "test_brax_problem_init"
+    "test_brax_problem_sample"
+    "test_gymnax_problem_eval"
+    "test_gymnax_problem_init"
+    "test_gymnax_problem_sample"
+    "test_torchvision_problem_eval"
+    "test_torchvision_problem_init"
+    "test_torchvision_problem_sample"
     "test_vision_fitness"
   ];