Fix ref

Lightning-Universe · Sep 10, 2021 · a1cf01f · a1cf01f
1 parent a1189fe
commit a1cf01f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_detection.py b/tests/models/test_detection.py
@@ -49,7 +49,7 @@ def test_fasterrcnn_pyt_module_bbone_train(tmpdir):
     train_dl = DataLoader(DummyDetectionDataset(), collate_fn=_collate_fn)
     valid_dl = DataLoader(DummyDetectionDataset(), collate_fn=_collate_fn)
 
-    trainer = pl.Trainer(fast_dev_run=True, default_root_dir=tmpdir)
+    trainer = Trainer(fast_dev_run=True, default_root_dir=tmpdir)
     trainer.fit(model, train_dl, valid_dl)