From 3a9f945f25709343f7985633439a16c0df2302a7 Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@zalando.de>
Date: Thu, 18 Jul 2019 12:19:58 +0200
Subject: [PATCH] import loader classes

---
 pts/dataset/__init__.py | 1 +
 1 file changed, 1 insertion(+)

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