Skip to content

Commit 2b4bcc7

Browse files
committed
Added TensorLoader import to __init__.py in data module. (#222)
1 parent b5732b2 commit 2b4bcc7

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

mipcandy/data/__init__.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from mipcandy.data.convertion import convert_ids_to_logits, convert_logits_to_ids, auto_convert
2-
from mipcandy.data.dataset import Loader, UnsupervisedDataset, SupervisedDataset, DatasetFromMemory, MergedDataset, \
3-
PathBasedUnsupervisedDataset, SimpleDataset, PathBasedSupervisedDataset, NNUNetDataset, BinarizedDataset
2+
from mipcandy.data.dataset import Loader, TensorLoader, UnsupervisedDataset, SupervisedDataset, DatasetFromMemory, \
3+
MergedDataset, PathBasedUnsupervisedDataset, SimpleDataset, PathBasedSupervisedDataset, NNUNetDataset, \
4+
BinarizedDataset
45
from mipcandy.data.download import download_dataset
56
from mipcandy.data.geometric import ensure_num_dimensions, orthographic_views, aggregate_orthographic_views, crop
67
from mipcandy.data.inspection import InspectionAnnotation, InspectionAnnotations, load_inspection_annotations, \

0 commit comments

Comments
 (0)