Commit b53d93de authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python311Packages.torchrl: 0.3.1 -> 0.4.0 

Diff: https://github.com/pytorch/rl/compare/refs/tags/v0.3.1...v0.4.0

Changelog: https://github.com/pytorch/rl/releases/tag/v0.4.0
parent d083724a

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

+29 −5
Original line number Diff line number Diff line
@@ -12,13 +12,22 @@ 
, ale-py

, gym

, pygame

, torchsnapshot

, gymnasium

, mujoco

, h5py

, huggingface-hub

, minari

, pandas

, pillow

, requests

, scikit-learn

, torchvision

, tqdm

, moviepy

, git

, hydra-core

, tensorboard

, tqdm

, wandb

, packaging

, tensordict
@@ -31,7 +40,7 @@ 


buildPythonPackage rec {

  pname = "torchrl";

  version = "0.3.1";

  version = "0.4.0";

  pyproject = true;



  disabled = pythonOlder "3.8";
@@ -40,17 +49,17 @@ buildPythonPackage rec { 
    owner = "pytorch";

    repo = "rl";

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

    hash = "sha256-lETW996IKPUGgZpe+cyzrXvVmDSwj5G4XFreFmGxReQ=";

    hash = "sha256-8wSyyErqveP9zZS/UGvWVBYyylu9BuA447GEjXIzBIk=";

  };



  nativeBuildInputs = [

  build-system = [

    ninja

    setuptools

    wheel

    which

  ];



  propagatedBuildInputs = [

  dependencies = [

    cloudpickle

    numpy

    packaging
@@ -64,10 +73,24 @@ buildPythonPackage rec { 
      gym

      pygame

    ];

    checkpointing = [

      torchsnapshot

    ];

    gym-continuous = [

      gymnasium

      mujoco

    ];

    offline-data = [

      h5py

      huggingface-hub

      minari

      pandas

      pillow

      requests

      scikit-learn

      torchvision

      tqdm

    ];

    rendering = [

      moviepy

    ];
@@ -103,6 +126,7 @@ buildPythonPackage rec { 
    pytestCheckHook

    pyyaml

    scipy

    torchvision

  ]

  ++ passthru.optional-dependencies.atari

  ++ passthru.optional-dependencies.gym-continuous