diff --git a/pts/dataset/__init__.py b/pts/dataset/__init__.py
index 7bff084..06ce12a 100644
--- a/pts/dataset/__init__.py
+++ b/pts/dataset/__init__.py
@@ -7,3 +7,4 @@ from pts.dataset.sampler import (
     BucketInstanceSampler,
 )
 from pts.dataset.sampler import InstanceSampler, UniformSplitSampler, TestSplitSampler, ExpectedNumInstanceSampler, BucketInstanceSampler
+from pts.dataset.loader import DataLoader, TrainDataLoader, InferenceDataLoader