amend

pytorch · Feb 5, 2024 · 42927e9 · 42927e9
1 parent bee9bf1
commit 42927e9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_tensordict.py b/test/test_tensordict.py
@@ -4156,6 +4156,7 @@ def test_squeeze_with_none(self, td_name, device, squeeze_dim=None):
             assert (td.get("a") == 1).all()
 
     @pytest.mark.filterwarnings("error")
+    @set_lazy_legacy(True)
     def test_stack_onto(self, td_name, device, tmpdir):
         torch.manual_seed(1)
         td = getattr(self, td_name)(device)