Skip to content

Commit 3dc08b1

Browse files
authored
Fix flaky test caused by weak reference (#14157)
1 parent a7cebf2 commit 3dc08b1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/tests_pytorch/trainer/connectors/test_data_connector.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -445,7 +445,8 @@ def test_dataloader_source_direct_access():
445445
def test_dataloader_source_request_from_module():
446446
"""Test requesting a dataloader from a module works."""
447447
module = BoringModel()
448-
module.trainer = Trainer()
448+
trainer = Trainer()
449+
module.trainer = trainer
449450
module.foo = Mock(return_value=module.train_dataloader())
450451

451452
source = _DataLoaderSource(module, "foo")

0 commit comments

Comments
 (0)