Skip to content

Commit 12b4d26

Browse files
committed
update
1 parent 85cbde1 commit 12b4d26

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/tests_fabric/strategies/test_ddp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def test_module_init_context(precision, expected_dtype):
146146
@mock.patch.dict(os.environ, {"LOCAL_RANK": "0"})
147147
@mock.patch("lightning.fabric.strategies.ddp.DistributedDataParallel")
148148
@mock.patch("torch.cuda.Stream")
149-
@mock.patch("torch.cuda.default_stream") # add this
149+
@mock.patch("torch.cuda.default_stream")
150150
@mock.patch("torch.cuda.is_current_stream_capturing", return_value=False)
151151
@mock.patch("torch.cuda.stream")
152152
def test_setup_with_cuda_stream(cuda_stream_mock, *_):

0 commit comments

Comments
 (0)