diff --git a/test/test_tensordict.py b/test/test_tensordict.py index be0ff5fef..fcd1626e1 100644 --- a/test/test_tensordict.py +++ b/test/test_tensordict.py @@ -621,9 +621,7 @@ def test_data_grad(self): assert td.data is not td.grad td.lock_() assert td.grad.is_locked - assert td.grad is td.grad assert td.data.is_locked - assert td.data is td.data @pytest.mark.parametrize( "stack_dim",