Skip to content

Commit

Permalink
python312Packages.torchrl: clean; skip tests failing because of gymna…
Browse files Browse the repository at this point in the history
…sium update
  • Loading branch information
GaetanLepage committed Oct 22, 2024
1 parent d5a44e3 commit 9552966
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions pkgs/development/python-modules/torchrl/default.nix
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
{
lib,
buildPythonPackage,
pythonOlder,
fetchFromGitHub,

# build-system
Expand Down Expand Up @@ -44,17 +43,13 @@
pytestCheckHook,
pyyaml,
scipy,

stdenv,
}:

buildPythonPackage rec {
pname = "torchrl";
version = "0.5.0";
pyproject = true;

disabled = pythonOlder "3.8";

src = fetchFromGitHub {
owner = "pytorch";
repo = "rl";
Expand Down Expand Up @@ -138,6 +133,11 @@ buildPythonPackage rec {
++ optional-dependencies.rendering;

disabledTests = [
# torchrl is incompatible with gymnasium>=1.0
# https://github.com/pytorch/rl/discussions/2483
"test_resetting_strategies"
"test_torchrl_to_gym"

# mujoco.FatalError: an OpenGL platform library has not been loaded into this process, this most likely means that a valid OpenGL context has not been created before mjr_makeContext was called
"test_vecenvs_env"

Expand Down Expand Up @@ -175,6 +175,5 @@ buildPythonPackage rec {
changelog = "https://github.com/pytorch/rl/releases/tag/v${version}";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ GaetanLepage ];
# ~3k tests fail with: RuntimeError: internal error
};
}

0 comments on commit 9552966

Please sign in to comment.