commit 23ff7d9dc4f3d553939e7bfe0d2667198f993536 · pyrox.dev/nixpkgs

+116 -101

pkgs/development/python-modules/bsuite/default.nix

···

       1
       -
       { lib

     

       2
       -
       , fetchPypi

     

       3
       -
       , buildPythonPackage

     

       4
       -
       , fetchpatch

     

       5
       -
       , frozendict

     

       6
       -
       , termcolor

     

       7
       -
       , matplotlib

     

       8
       -
       , gym

     

       9
       -
       , pandas

     

       10
       -
       , scipy

     

       11
       -
       , absl-py

     

       12
       -
       , dm-env

     

       13
       -
       , plotnine

     

       14
       -
       , scikit-image

     

       15
       -
       , dm-tree

     

       16
       -
       , patsy

     

       17
       -
       , tensorflow-probability

     

       18
       -
       , dm-haiku

     

       19
       -
       , statsmodels

     

       20
       -
       , mizani

     

       21
       -
       , trfl

     

       22
       -
       , optax

     

       23
       -
       , pytestCheckHook

     

       24
       -
       , dm-sonnet

     

       25
       -
       , rlax

     

       26
       -
       , distrax

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       27
        
       }:

     

       28
        
       

     

       29
       -
       let bsuite = buildPythonPackage rec {

     

       30
       -
         pname = "bsuite";

     

       31
       -
         version = "0.3.5";

     

       32
       -
         format = "setuptools";

     

       0
        
       
     

       33
        
       

     

       34
       -
         src = fetchPypi {

     

       35
       -
           inherit pname version;

     

       36
       -
           hash = "sha256-ak9McvXl7Nz5toUaPaRaJek9lurxiQiIW209GnZEjX0=";

     

       37
       -
         };

     

       38
        
       

     

       39
       -
         patches = [

     

       40
       -
           (fetchpatch {  # Convert np.int -> np.int32 since np.int is deprecated (https://github.com/google-deepmind/bsuite/pull/48)

     

       41
       -
             url = "https://github.com/google-deepmind/bsuite/pull/48/commits/f8d81b2f1c27ef2c8c71ae286001ed879ea306ab.patch";

     

       42
       -
             hash = "sha256-FXtvVS+U8brulq8Z27+yWIimB+kigGiUOIv1SHb1TA8=";

     

       43
       -
           })

     

       44
       -
         ];

     

       45
        
       

     

       46
       -
         propagatedBuildInputs = [

     

       47
       -
           absl-py

     

       48
       -
           dm-env

     

       49
       -
           dm-tree

     

       50
       -
           frozendict

     

       51
       -
           gym

     

       52
       -
           matplotlib

     

       53
       -
           mizani

     

       54
       -
           pandas

     

       55
       -
           patsy

     

       56
       -
           plotnine

     

       57
       -
           scikit-image

     

       58
       -
           scipy

     

       59
       -
           statsmodels

     

       60
       -
           termcolor

     

       61
       -
         ];

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       62
        
       

     

       63
       -
         nativeCheckInputs = [

     

       64
       -
           distrax

     

       65
       -
           dm-haiku

     

       66
       -
           dm-sonnet

     

       67
       -
           optax

     

       68
       -
           pytestCheckHook

     

       69
       -
           rlax

     

       70
       -
           tensorflow-probability

     

       71
       -
           trfl

     

       72
       -
         ];

     

       73
        
       

     

       74
       -
         pythonImportsCheck = [

     

       75
       -
           "bsuite"

     

       76
       -
         ];

     

       77
        
       

     

       78
       -
         disabledTests = [

     

       79
       -
           # Tests require network connection

     

       80
       -
           "test_run9"

     

       81
       -
           "test_longer_action_sequence"

     

       82
       -
           "test_reset"

     

       83
       -
           "test_step_after_reset"

     

       84
       -
           "test_step_on_fresh_environment"

     

       85
       -
           "test_longer_action_sequence"

     

       86
       -
           "test_reset"

     

       87
       -
           "test_step_after_reset"

     

       88
       -
           "test_step_on_fresh_environment"

     

       89
       -
           "test_longer_action_sequence"

     

       90
       -
           "test_reset"

     

       91
       -
           "test_step_after_reset"

     

       92
       -
           "test_step_on_fresh_environment"

     

       93
       -
           "test_logger"

     

       94
       -
           "test_episode_truncation"

     

       95
       -
         ];

     

       96
        
       

     

       97
       -
         # escape infinite recursion with rlax

     

       98
       -
         doCheck = false;

     

       99
        
       

     

       100
       -
         passthru.tests = {

     

       101
       -
           check = bsuite.overridePythonAttrs (_: { doCheck = true; });

     

       102
       -
         };

     

       0
        
       
     

       0
        
       
     

       103
        
       

     

       104
       -
         meta = with lib; {

     

       105
       -
           description = ''

     

       106
       -
             Core RL Behaviour Suite. A collection of reinforcement learning

     

       107
       -
             experiments

     

       108
       -
           '';

     

       109
       -
           homepage = "https://github.com/deepmind/bsuite";

     

       110
       -
           license = licenses.asl20;

     

       111
       -
           maintainers = with maintainers; [ onny ];

     

       112
        
         };

     

       113
       -
       }; in bsuite

     

       0

···

       1
       +
       {

     

       2
       +
         lib,

     

       3
       +
         absl-py,

     

       4
       +
         buildPythonPackage,

     

       5
       +
         distrax,

     

       6
       +
         dm-env,

     

       7
       +
         dm-haiku,

     

       8
       +
         dm-sonnet,

     

       9
       +
         dm-tree,

     

       10
       +
         fetchpatch,

     

       11
       +
         fetchPypi,

     

       12
       +
         frozendict,

     

       13
       +
         gym,

     

       14
       +
         matplotlib,

     

       15
       +
         mizani,

     

       16
       +
         optax,

     

       17
       +
         pandas,

     

       18
       +
         patsy,

     

       19
       +
         plotnine,

     

       20
       +
         pytestCheckHook,

     

       21
       +
         pythonOlder,

     

       22
       +
         rlax,

     

       23
       +
         scikit-image,

     

       24
       +
         scipy,

     

       25
       +
         setuptools,

     

       26
       +
         statsmodels,

     

       27
       +
         tensorflow-probability,

     

       28
       +
         termcolor,

     

       29
       +
         trfl,

     

       30
        
       }:

     

       31
        
       

     

       32
       +
       let

     

       33
       +
         bsuite = buildPythonPackage rec {

     

       34
       +
           pname = "bsuite";

     

       35
       +
           version = "0.3.5";

     

       36
       +
           pyproject = true;

     

       37
        
       

     

       38
       +
           disabled = pythonOlder "3.7";

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       39
        
       

     

       40
       +
           src = fetchPypi {

     

       41
       +
             inherit pname version;

     

       42
       +
             hash = "sha256-ak9McvXl7Nz5toUaPaRaJek9lurxiQiIW209GnZEjX0=";

     

       43
       +
           };

     

       0
        
       
     

       0
        
       
     

       44
        
       

     

       45
       +
           patches = [

     

       46
       +
             # Convert np.int -> np.int32 since np.int is deprecated, https://github.com/google-deepmind/bsuite/pull/48

     

       47
       +
             (fetchpatch {

     

       48
       +
               url = "https://github.com/google-deepmind/bsuite/pull/48/commits/f8d81b2f1c27ef2c8c71ae286001ed879ea306ab.patch";

     

       49
       +
               hash = "sha256-FXtvVS+U8brulq8Z27+yWIimB+kigGiUOIv1SHb1TA8=";

     

       50
       +
             })

     

       51
       +
             # Replace imp with importlib, https://github.com/google-deepmind/bsuite/pull/50

     

       52
       +
             (fetchpatch {

     

       53
       +
               name = "replace-imp.patch";

     

       54
       +
               url = "https://github.com/google-deepmind/bsuite/commit/d08b63655c7efa5b5bb0f35e825e17549d23e812.patch";

     

       55
       +
               hash = "sha256-V5p/6edNXTpEckuSuxJ/mvfJng5yE/pfeMoYbvlNpEo=";

     

       56
       +
             })

     

       57
       +
           ];

     

       58
       +
       

     

       59
       +
           build-system = [ setuptools ];

     

       60
       +
       

     

       61
       +
           dependencies = [

     

       62
       +
             absl-py

     

       63
       +
             dm-env

     

       64
       +
             dm-tree

     

       65
       +
             frozendict

     

       66
       +
             gym

     

       67
       +
             matplotlib

     

       68
       +
             mizani

     

       69
       +
             pandas

     

       70
       +
             patsy

     

       71
       +
             plotnine

     

       72
       +
             scikit-image

     

       73
       +
             scipy

     

       74
       +
             statsmodels

     

       75
       +
             termcolor

     

       76
       +
           ];

     

       77
        
       

     

       78
       +
           nativeCheckInputs = [

     

       79
       +
             distrax

     

       80
       +
             dm-haiku

     

       81
       +
             dm-sonnet

     

       82
       +
             optax

     

       83
       +
             pytestCheckHook

     

       84
       +
             rlax

     

       85
       +
             tensorflow-probability

     

       86
       +
             trfl

     

       87
       +
           ];

     

       88
        
       

     

       89
       +
           pythonImportsCheck = [ "bsuite" ];

     

       0
        
       
     

       0
        
       
     

       90
        
       

     

       91
       +
           disabledTests = [

     

       92
       +
             # Tests require network connection

     

       93
       +
             "test_run9"

     

       94
       +
             "test_longer_action_sequence"

     

       95
       +
             "test_reset"

     

       96
       +
             "test_step_after_reset"

     

       97
       +
             "test_step_on_fresh_environment"

     

       98
       +
             "test_longer_action_sequence"

     

       99
       +
             "test_reset"

     

       100
       +
             "test_step_after_reset"

     

       101
       +
             "test_step_on_fresh_environment"

     

       102
       +
             "test_longer_action_sequence"

     

       103
       +
             "test_reset"

     

       104
       +
             "test_step_after_reset"

     

       105
       +
             "test_step_on_fresh_environment"

     

       106
       +
             "test_logger"

     

       107
       +
             "test_episode_truncation"

     

       108
       +
           ];

     

       109
        
       

     

       110
       +
           # Escape infinite recursion with rlax

     

       111
       +
           doCheck = false;

     

       112
        
       

     

       113
       +
           passthru.tests = {

     

       114
       +
             check = bsuite.overridePythonAttrs (_: {

     

       115
       +
               doCheck = true;

     

       116
       +
             });

     

       117
       +
           };

     

       118
        
       

     

       119
       +
           meta = with lib; {

     

       120
       +
             description = "Collection of experiments that investigate core capabilities of a reinforcement learning (RL) agent";

     

       121
       +
             homepage = "https://github.com/deepmind/bsuite";

     

       122
       +
             changelog = "https://github.com/google-deepmind/bsuite/releases/tag/${version}";

     

       123
       +
             license = licenses.asl20;

     

       124
       +
             maintainers = with maintainers; [ onny ];

     

       125
       +
           };

     

       0
        
       
     

       126
        
         };

     

       127
       +
       in

     

       128
       +
       bsuite

+23 -26

pkgs/development/python-modules/plotnine/default.nix

···

       1
       -
       { lib

     

       2
       -
       , buildPythonPackage

     

       3
       -
       , pythonOlder

     

       4
       -
       , fetchFromGitHub

     

       5
       -
       , setuptools-scm

     

       6
       -
       , matplotlib

     

       7
       -
       , mizani

     

       8
       -
       , pandas

     

       9
       -
       , patsy

     

       10
       -
       , scipy

     

       11
       -
       , statsmodels

     

       12
       -
       , geopandas

     

       13
       -
       , pytestCheckHook

     

       14
       -
       , scikit-misc

     

       0
        
       
     

       15
        
       }:

     

       16
        
       

     

       17
        
       buildPythonPackage rec {

     

       18
        
         pname = "plotnine";

     

       19
       -
         version = "0.13.3";

     

       20
        
         pyproject = true;

     

       21
        
       

     

       22
        
         disabled = pythonOlder "3.9";

     
···

       25
        
           owner = "has2k1";

     

       26
        
           repo = "plotnine";

     

       27
        
           rev = "refs/tags/v${version}";

     

       28
       -
           hash = "sha256-dbfbXYYmVdufTtrrllrqwe87LL1nYRar4RMLef7ajTQ=";

     

       29
        
         };

     

       30
       -
       

     

       31
       -
         nativeBuildInputs = [

     

       32
       -
           setuptools-scm

     

       33
       -
         ];

     

       34
        
       

     

       35
        
         postPatch = ''

     

       36
        
           substituteInPlace pyproject.toml \

     

       37
       -
             --replace " --cov=plotnine --cov-report=xml" ""

     

       38
        
         '';

     

       39
        
       

     

       40
       -
         propagatedBuildInputs = [

     

       0
        
       
     

       0
        
       
     

       41
        
           matplotlib

     

       42
        
           mizani

     

       43
        
           pandas

     
···

       56
        
           export HOME=$(mktemp -d)

     

       57
        
         '';

     

       58
        
       

     

       59
       -
         pythonImportsCheck = [

     

       60
       -
           "plotnine"

     

       61
       -
         ];

     

       62
        
       

     

       63
        
         disabledTestPaths = [

     

       64
        
           # Assertion Errors:

     
···

       103
        
         ];

     

       104
        
       

     

       105
        
         meta = with lib; {

     

       106
       -
           description = "Grammar of graphics for python";

     

       107
        
           homepage = "https://plotnine.readthedocs.io/";

     

       108
        
           changelog = "https://github.com/has2k1/plotnine/releases/tag/v${version}";

     

       109
        
           license = licenses.mit;

···

       1
       +
       {

     

       2
       +
         lib,

     

       3
       +
         buildPythonPackage,

     

       4
       +
         fetchFromGitHub,

     

       5
       +
         geopandas,

     

       6
       +
         matplotlib,

     

       7
       +
         mizani,

     

       8
       +
         pandas,

     

       9
       +
         patsy,

     

       10
       +
         pytestCheckHook,

     

       11
       +
         pythonOlder,

     

       12
       +
         scikit-misc,

     

       13
       +
         scipy,

     

       14
       +
         setuptools-scm,

     

       15
       +
         statsmodels,

     

       16
        
       }:

     

       17
        
       

     

       18
        
       buildPythonPackage rec {

     

       19
        
         pname = "plotnine";

     

       20
       +
         version = "0.13.4";

     

       21
        
         pyproject = true;

     

       22
        
       

     

       23
        
         disabled = pythonOlder "3.9";

     
···

       26
        
           owner = "has2k1";

     

       27
        
           repo = "plotnine";

     

       28
        
           rev = "refs/tags/v${version}";

     

       29
       +
           hash = "sha256-ylsaV5yWVbxvD74spAI5tDwIjjue7MOMaGgp4Dc8Nhk=";

     

       30
        
         };

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       31
        
       

     

       32
        
         postPatch = ''

     

       33
        
           substituteInPlace pyproject.toml \

     

       34
       +
             --replace-fail " --cov=plotnine --cov-report=xml" ""

     

       35
        
         '';

     

       36
        
       

     

       37
       +
         build-system = [ setuptools-scm ];

     

       38
       +
       

     

       39
       +
         dependencies = [

     

       40
        
           matplotlib

     

       41
        
           mizani

     

       42
        
           pandas

     
···

       55
        
           export HOME=$(mktemp -d)

     

       56
        
         '';

     

       57
        
       

     

       58
       +
         pythonImportsCheck = [ "plotnine" ];

     

       0
        
       
     

       0
        
       
     

       59
        
       

     

       60
        
         disabledTestPaths = [

     

       61
        
           # Assertion Errors:

     
···

       100
        
         ];

     

       101
        
       

     

       102
        
         meta = with lib; {

     

       103
       +
           description = "Grammar of graphics for Python";

     

       104
        
           homepage = "https://plotnine.readthedocs.io/";

     

       105
        
           changelog = "https://github.com/has2k1/plotnine/releases/tag/v${version}";

     

       106
        
           license = licenses.mit;