tests

facebookresearch · Jul 31, 2024 · 6e33e7e · 6e33e7e
1 parent cefb586
commit 6e33e7e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/test/test_pettingzoo.py b/test/test_pettingzoo.py
@@ -113,7 +113,7 @@ def test_gnn(
     @pytest.mark.parametrize("task", [PettingZooTask.SIMPLE_TAG])
     @pytest.mark.skipif(
         packaging.version.parse(torchrl.__version__).local is None,
-        "gru model needs torchrl from github",
+        reason="gru model needs torchrl from github",
     )
     def test_gru(
         self,

diff --git a/test/test_smacv2.py b/test/test_smacv2.py
@@ -83,7 +83,7 @@ def test_gnn(
     @pytest.mark.parametrize("task", [Smacv2Task.PROTOSS_5_VS_5])
     @pytest.mark.skipif(
         packaging.version.parse(torchrl.__version__).local is None,
-        "gru model needs torchrl from github",
+        reason="gru model needs torchrl from github",
     )
     def test_gru(
         self,

diff --git a/test/test_vmas.py b/test/test_vmas.py
@@ -122,7 +122,7 @@ def test_gnn(
     @pytest.mark.parametrize("task", [VmasTask.NAVIGATION])
     @pytest.mark.skipif(
         packaging.version.parse(torchrl.__version__).local is None,
-        "gru model needs torchrl from github",
+        reason="gru model needs torchrl from github",
     )
     def test_gru(
         self,