@@ -2,5 +2,6 @@
 
 from tests.integration_tests.ema_train_integration_test import EMAIntegrationTest
 from tests.integration_tests.lr_test import LRTest
+from tests.integration_tests.pose_estimation_dataset_test import PoseEstimationDatasetIntegrationTest
 
-__all__ = ["EMAIntegrationTest", "LRTest"]
+__all__ = ["EMAIntegrationTest", "LRTest", "PoseEstimationDatasetIntegrationTest"]