From 3405a45c9d024b6b5e0bf829f918b87a4bd57fc6 Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Fri, 2 Aug 2024 10:41:32 +0200 Subject: [PATCH] test --- test/test_models.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/test/test_models.py b/test/test_models.py index eb71fc89..53dfae0c 100644 --- 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)