Skip to content

Commit 294b10b

Browse files
committed
fix(test): use correct kwarg cleanup_previous instead of needs_cleanup
(cherry picked from commit 402ad07)
1 parent 1d85c3a commit 294b10b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/registered/debug_utils/test_dumper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -564,7 +564,7 @@ def test_cleanup_removes_old_dumps(self, tmp_path):
564564
old_dir.mkdir()
565565
(old_dir / "dummy.pt").touch()
566566

567-
dumper = _make_test_dumper(tmp_path, needs_cleanup=True)
567+
dumper = _make_test_dumper(tmp_path, cleanup_previous=True)
568568
dumper.dump("new_tensor", torch.randn(3, 3))
569569

570570
assert not old_dir.exists()

0 commit comments

Comments
 (0)