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

python312Packages.torchrl: skip flaky test on aarch64-linux

parent 9d87b111

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

+46 −40
Original line number Diff line number Diff line 
{

  lib,

  stdenv,

  buildPythonPackage,

  fetchFromGitHub,
@@ -132,7 +133,8 @@ buildPythonPackage rec { 
    ++ optional-dependencies.gym-continuous

    ++ optional-dependencies.rendering;



  disabledTests = [

  disabledTests =

    [

      # torchrl is incompatible with gymnasium>=1.0

      # https://github.com/pytorch/rl/discussions/2483

      "test_resetting_strategies"
@@ -171,6 +173,10 @@ buildPythonPackage rec { 
      # Flaky (hangs indefinitely on some CPUs)

      "test_gae_multidim"

      "test_gae_param_as_tensor"

    ]

    ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [

      # Flaky (AssertionError)

      "test_vecnorm_parallel_auto"

    ];



  meta = {