LSTMModule test

pytorch · Nov 29, 2023 · 5910811 · 5910811
1 parent 311078d
commit 5910811
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/test_tensordictmodules.py b/test/test_tensordictmodules.py
@@ -1672,8 +1672,7 @@ def test_noncontiguous(self):
         lstm_module(padded)
 
     @pytest.mark.parametrize("shape", [[], [2], [2, 3], [2, 3, 4]])
-    @pytest.mark.parametrize("python_based", [True, False])
-    def test_singel_step(self, shape, python_based):
+    def test_single_step(self, shape, python_based):
         td = TensorDict(
             {
                 "observation": torch.zeros(*shape, 3),