commit dfbce9a4d45d67edf7f7add221f6c5133f6b12ea · pyrox.dev/nixpkgs

+44 -7
pkgs/development/python-modules/torchrl/default.nix
···

       5
       5
        
         fetchFromGitHub,

     

       6
       6
        
       

     

       7
       7
        
         # build-system

     

       8
       8
       +
         cmake,

     

       8
       9
        
         ninja,

     

       10
       10
       +
         numpy,

     

       11
       11
       +
         pybind11,

     

       9
       12
        
         setuptools,

     

       10
       10
       -
         which,

     

       13
       13
       +
         torch,

     

       11
       14
        
       

     

       12
       15
        
         # dependencies

     

       13
       16
        
         cloudpickle,

     

       14
       14
       -
         numpy,

     

       15
       17
        
         packaging,

     

       18
       18
       +
         pyvers,

     

       16
       19
        
         tensordict,

     

       17
       17
       -
         torch,

     

       18
       20
        
       

     

       19
       21
        
         # optional-dependencies

     

       20
       22
        
         # atari

     

       21
       23
        
         gymnasium,

     

       24
       24
       +
         # brax

     

       25
       25
       +
         brax,

     

       26
       26
       +
         jax,

     

       22
       27
        
         # checkpointing

     

       23
       28
        
         torchsnapshot,

     

       29
       29
       +
         # dm-control

     

       30
       30
       +
         dm-control,

     

       24
       31
        
         # gym-continuous

     

       25
       32
        
         mujoco,

     

       26
       33
        
         # llm

     
···

       36
       43
        
         sentencepiece,

     

       37
       44
        
         transformers,

     

       38
       45
        
         vllm,

     

       46
       46
       +
         # marl

     

       47
       47
       +
         pettingzoo,

     

       39
       48
        
         # offline-data

     

       40
       49
        
         h5py,

     

       41
       50
        
         huggingface-hub,

     
···

       64
       73
        
       

     

       65
       74
        
       buildPythonPackage rec {

     

       66
       75
        
         pname = "torchrl";

     

       67
       67
       -
         version = "0.9.2";

     

       76
       76
       +
         version = "0.10.0";

     

       68
       77
        
         pyproject = true;

     

       69
       78
        
       

     

       70
       79
        
         src = fetchFromGitHub {

     

       71
       80
        
           owner = "pytorch";

     

       72
       81
        
           repo = "rl";

     

       73
       82
        
           tag = "v${version}";

     

       74
       74
       -
           hash = "sha256-6rU5+J70T0E7+60jihsjwlLls8jJlxKi3nmrL0xm2c0=";

     

       83
       83
       +
           hash = "sha256-DqLB1JnQ96cxVEzcXra1hFVfrN7eXTlTwPtlPClnaBA=";

     

       75
       84
        
         };

     

       76
       85
        
       

     

       86
       86
       +
         postPatch = ''

     

       87
       87
       +
           substituteInPlace pyproject.toml \

     

       88
       88
       +
             --replace-fail "pybind11[global]" "pybind11"

     

       89
       89
       +
         '';

     

       90
       90
       +
       

     

       77
       91
        
         build-system = [

     

       92
       92
       +
           cmake

     

       78
       93
        
           ninja

     

       94
       94
       +
           numpy

     

       95
       95
       +
           pybind11

     

       79
       96
        
           setuptools

     

       80
       80
       -
           which

     

       97
       97
       +
           torch

     

       81
       98
        
         ];

     

       99
       99
       +
         dontUseCmakeConfigure = true;

     

       82
       100
        
       

     

       83
       101
        
         dependencies = [

     

       84
       102
        
           cloudpickle

     

       85
       103
        
           numpy

     

       86
       104
        
           packaging

     

       87
       105
        
           tensordict

     

       106
       106
       +
           pyvers

     

       88
       107
        
           torch

     

       89
       108
        
         ];

     

       90
       109
        
       

     

       91
       110
        
         optional-dependencies = {

     

       92
       92
       -
           atari = gymnasium.optional-dependencies.atari;

     

       111
       111
       +
           atari = [

     

       112
       112
       +
             gymnasium

     

       113
       113
       +
           ]

     

       114
       114
       +
           ++ gymnasium.optional-dependencies.atari;

     

       115
       115
       +
           brax = [

     

       116
       116
       +
             brax

     

       117
       117
       +
             jax

     

       118
       118
       +
           ];

     

       93
       119
        
           checkpointing = [ torchsnapshot ];

     

       120
       120
       +
           dm-control = [ dm-control ];

     

       94
       121
        
           gym-continuous = [

     

       95
       122
        
             gymnasium

     

       96
       123
        
             mujoco

     
···

       109
       136
        
             transformers

     

       110
       137
        
             vllm

     

       111
       138
        
           ];

     

       139
       139
       +
           marl = [

     

       140
       140
       +
             # dm-meltingpot (unpackaged)

     

       141
       141
       +
             pettingzoo

     

       142
       142
       +
             # vmas (unpackaged)

     

       143
       143
       +
           ];

     

       112
       144
        
           offline-data = [

     

       113
       145
        
             h5py

     

       114
       146
        
             huggingface-hub

     
···

       120
       152
        
             torchvision

     

       121
       153
        
             tqdm

     

       122
       154
        
           ];

     

       155
       155
       +
           open-spiel = [

     

       156
       156
       +
             # open-spiel (unpackaged)

     

       157
       157
       +
           ];

     

       123
       158
        
           rendering = [ moviepy ];

     

       159
       159
       +
           replay-buffer = [ torch ];

     

       124
       160
        
           utils = [

     

       125
       161
        
             git

     

       126
       162
        
             hydra-core

     

       163
       163
       +
             # hydra-submitit-launcher (unpackaged)

     

       127
       164
        
             tensorboard

     

       128
       165
        
             tqdm

     

       129
       166
        
             wandb