@@ -1,7 +1,6 @@
 # PACKAGE IMPORTS FOR EXTERNAL USAGE
 
-from tests.integration_tests.s3_dataset_test import TestDataset
 from tests.integration_tests.ema_train_integration_test import EMAIntegrationTest
 from tests.integration_tests.lr_test import LRTest
 
-_all__ = [TestDataset, EMAIntegrationTest, LRTest]
+__all__ = ["EMAIntegrationTest", "LRTest"]