commit ce75ef61f9686bf14580f23126ec5be614676181 · pyrox.dev/nixpkgs

+11 -5
pkgs/development/python-modules/evosax/default.nix
···

       27
       27
        
       

     

       28
       28
        
       buildPythonPackage rec {

     

       29
       29
        
         pname = "evosax";

     

       30
       30
       -
         version = "0.1.6";

     

       30
       30
       +
         version = "0.2.0";

     

       31
       31
        
         pyproject = true;

     

       32
       32
        
       

     

       33
       33
        
         src = fetchFromGitHub {

     

       34
       34
        
           owner = "RobertTLange";

     

       35
       35
        
           repo = "evosax";

     

       36
       36
        
           tag = "v.${version}";

     

       37
       37
       -
           hash = "sha256-v8wRiWZlJPF9pIXocQ6/caHl1W4QBNjkmuImJ6MAueo=";

     

       37
       37
       +
           hash = "sha256-ye5IHM8Pn/+BXI9kcB3W281Gna9hXV8DwsaJ9Xu06fU=";

     

       38
       38
        
         };

     

       39
       39
        
       

     

       40
       40
        
         build-system = [ setuptools ];

     

       41
       41
        
       

     

       42
       42
        
         dependencies = [

     

       43
       43
       -
           chex

     

       44
       43
        
           dotmap

     

       45
       44
        
           flax

     

       46
       45
        
           jax

     

       47
       47
       -
           jaxlib

     

       48
       46
        
           matplotlib

     

       49
       47
        
           numpy

     

       50
       50
       -
           pyyaml

     

       51
       48
        
         ];

     

       52
       49
        
       

     

       53
       50
        
         pythonImportsCheck = [ "evosax" ];

     
···

       66
       63
        
           "test_env_ffw_rollout"

     

       67
       64
        
       

     

       68
       65
        
           # Tries to download a data set from the internet

     

       66
       66
       +
           "test_brax_problem_eval"

     

       67
       67
       +
           "test_brax_problem_init"

     

       68
       68
       +
           "test_brax_problem_sample"

     

       69
       69
       +
           "test_gymnax_problem_eval"

     

       70
       70
       +
           "test_gymnax_problem_init"

     

       71
       71
       +
           "test_gymnax_problem_sample"

     

       72
       72
       +
           "test_torchvision_problem_eval"

     

       73
       73
       +
           "test_torchvision_problem_init"

     

       74
       74
       +
           "test_torchvision_problem_sample"

     

       69
       75
        
           "test_vision_fitness"

     

       70
       76
        
         ];

     

       71
       77