Skip to content

Commit

Permalink
[BugFix] Fix grad tests
Browse files Browse the repository at this point in the history
ghstack-source-id: 32c9ee0690db01b54de2e9cc0d83b595b87ac527
Pull Request resolved: #1070
  • Loading branch information
vmoens committed Nov 1, 2024
1 parent e64a4c3 commit 752e6dc
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions test/test_tensordict.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit 752e6dc

Please sign in to comment.