clean up

Lightning-AI · Sep 29, 2021 · a10c526 · a10c526
1 parent 6d6b3ea
commit a10c526
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 13 deletions.
diff --git a/pytorch_lightning/plugins/training_type/tpu_spawn.py b/pytorch_lightning/plugins/training_type/tpu_spawn.py
@@ -27,7 +27,6 @@
 from pytorch_lightning.overrides import LightningDistributedModule
 from pytorch_lightning.plugins.io.checkpoint_plugin import CheckpointIO
 from pytorch_lightning.plugins.training_type.ddp_spawn import DDPSpawnPlugin
-from pytorch_lightning.trainer.connectors.data_connector import _PatchDataLoader
 from pytorch_lightning.trainer.states import TrainerFn
 from pytorch_lightning.utilities import _OMEGACONF_AVAILABLE, _TPU_AVAILABLE, rank_zero_warn
 from pytorch_lightning.utilities.apply_func import apply_to_collection

diff --git a/tests/utilities/test_model_helpers.py b/tests/utilities/test_model_helpers.py
@@ -77,15 +77,3 @@ def bar(self):
     # `partial` support
     model.training_step = partial(model.training_step)
     assert is_overridden("training_step", model)
-
-    # TODO: remove
-    # `_PatchDataLoader.patch_loader_code` support
-    class TestModel(BoringModel):
-        def on_fit_start(self):
-            assert is_overridden("train_dataloader", self)
-            self.on_fit_start_called = True
-
-    model = TestModel()
-    trainer = Trainer(fast_dev_run=1)
-    trainer.fit(model, train_dataloader=model.train_dataloader())
-    assert model.on_fit_start_called