Skip to content

Commit 292c640

Browse files
pre-commit-ci[bot]four4fish
authored andcommitted
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent a74f4c1 commit 292c640

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/accelerators/test_cpu.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,7 @@
1717
def test_restore_checkpoint_after_pre_dispatch_default():
1818
"""Assert default for restore_checkpoint_after_pre_dispatch is False."""
1919
plugin = SingleDevicePlugin(
20-
accelerator=CPUAccelerator(),
21-
device=torch.device("cpu"),
22-
precision_plugin=PrecisionPlugin()
20+
accelerator=CPUAccelerator(), device=torch.device("cpu"), precision_plugin=PrecisionPlugin()
2321
)
2422
assert not plugin.restore_checkpoint_after_pre_dispatch
2523

@@ -55,7 +53,7 @@ def load_checkpoint(self, checkpoint_path: Union[str, Path]) -> Dict[str, Any]:
5553
accelerator=CPUAccelerator(),
5654
precision_plugin=PrecisionPlugin(),
5755
device=torch.device("cpu"),
58-
checkpoint_io=TorchCheckpointIO()
56+
checkpoint_io=TorchCheckpointIO(),
5957
)
6058
assert plugin.restore_checkpoint_after_pre_dispatch == restore_after_pre_dispatch
6159

0 commit comments

Comments
 (0)