test

facebookresearch · Aug 2, 2024 · 3405a45 · 3405a45
1 parent 345a5bd
commit 3405a45
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/test/test_models.py b/test/test_models.py
@@ -163,11 +163,6 @@ def test_models_forward_shape(
         or (isinstance(model_name, list) and model_name[0] != "gnn")
     ):
         pytest.skip("gnn model needs agent dim as input")
-    if (
-        packaging.version.parse(torchrl.__version__).local is None
-        and "gru" in model_name
-    ):
-        pytest.skip("gru model needs torchrl from github")
 
     torch.manual_seed(0)