GRUModule test

pytorch · Nov 29, 2023 · 1f2a586 · 1f2a586
1 parent 0a771f3
commit 1f2a586
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/test_tensordictmodules.py b/test/test_tensordictmodules.py
@@ -1775,6 +1775,7 @@ def test_multi_consecutive(self, shape, python_based):
             lstm_module_ss(td_ss)
             td_ss = step_mdp(td_ss, keep_other=True)
             td_ss["observation"][:] = _t + 1
+        import ipdb; ipdb.set_trace()  # asssert fails
         torch.testing.assert_close(
             td_ss["intermediate"], td["intermediate"][..., -1, :]
         )
@@ -2045,6 +2046,7 @@ def test_multi_consecutive(self, shape, python_based):
             gru_module_ss(td_ss)
             td_ss = step_mdp(td_ss, keep_other=True)
             td_ss["observation"][:] = _t + 1
+        import ipdb; ipdb.set_trace()  # asssert fails
         torch.testing.assert_close(
             td_ss["intermediate"], td["intermediate"][..., -1, :]
         )