Remove redundant code.

Use data wrappers in model wrapper APIs.
More typing.
This commit is contained in:
abigailt 2022-03-06 21:15:07 +02:00
parent 9f4d649934
commit 3d82db80c4
5 changed files with 57 additions and 166 deletions

View file

@ -3,4 +3,4 @@ The AI Privacy Toolbox (datasets).
Implementation of datasets utility components for datasets creation, load, and store
"""
from apt.utils.datasets.datasets import DatasetABC, StoredDatasetABC, DatasetFactory, Data, BaseDataset
from apt.utils.datasets.datasets import DatasetABC, StoredDatasetABC, DatasetFactory, Data, BaseDataset, DATA_ARRAY_TYPE