We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7d1cc5c commit f759600Copy full SHA for f759600
tests/tests_fabric/plugins/io/test_distributed_async_io.py
@@ -32,10 +32,10 @@ def test_async_checkpointio_save_options_forwarded(tmp_path):
32
dcp_kwargs = kwargs["dcp_kwargs"]
33
assert dcp_kwargs["foo"] == 123
34
if _TORCH_GREATER_EQUAL_2_9:
35
- assert "planner" in dcp_kwargs
+ assert "no_dist" in dcp_kwargs
36
if _TORCH_GREATER_EQUAL_2_7:
37
+ assert "planner" in dcp_kwargs
38
assert "async_checkpointer_type" in dcp_kwargs
- assert "no_dist" in dcp_kwargs
39
40
41
@RunIf(min_torch="2.4")
0 commit comments