Apply new importing

Lightning-Universe · Jan 4, 2021 · 0c9966f · 0c9966f
1 parent edd09b5
commit 0c9966f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pl_bolts/datasets/cifar10_dataset.py b/pl_bolts/datasets/cifar10_dataset.py
@@ -6,7 +6,7 @@
 import torch
 from torch import Tensor
 
-from pl_bolts.datasets.base_dataset import LightDataset
+from pl_bolts.datasets import LightDataset
 from pl_bolts.utils import _PIL_AVAILABLE
 from pl_bolts.utils.warnings import warn_missing_pkg
 

diff --git a/pl_bolts/models/self_supervised/amdim/datasets.py b/pl_bolts/models/self_supervised/amdim/datasets.py
@@ -2,8 +2,8 @@
 
 from torch.utils.data import random_split
 
-from pl_bolts.datasets.imagenet_dataset import UnlabeledImagenet
-from pl_bolts.datasets.ssl_amdim_datasets import CIFAR10Mixed
+from pl_bolts.datasets import UnlabeledImagenet
+from pl_bolts.datasets import CIFAR10Mixed
 from pl_bolts.models.self_supervised.amdim import transforms as amdim_transforms
 from pl_bolts.utils import _TORCHVISION_AVAILABLE
 from pl_bolts.utils.warnings import warn_missing_pkg