fix test

pytorch · Sep 27, 2024 · 6670d0a · 6670d0a
1 parent 1b26ad4
commit 6670d0a
Showing 1 changed file with 0 additions and 12 deletions.
diff --git a/test/stateful_dataloader/test_state_dict.py b/test/stateful_dataloader/test_state_dict.py
@@ -918,18 +918,6 @@ def test_num_workers_mismatch(self):
             self.assertTrue(False, "Error should be of type AssertionError")
 
 
-class TestTorchDataLazyImport_shard3(TestCase):
-    def test_lazy_imports(self) -> None:
-        import torchdata
-
-        self.assertFalse("datapipes" in torchdata.__dict__)
-
-        from torchdata import datapipes as dp, janitor  # noqa
-
-        self.assertTrue("datapipes" in torchdata.__dict__)
-        dp.iter.IterableWrapper([1, 2])
-
-
 class TestConcurrentDataLoaders_shard3(TestCase):
     def test_two_dataloaders(self) -> None:
         dataset = DummyMapDataset(100, shuffle=False)