# coding=utf-8
# Copyright 2023 DeepMind Technologies Limited.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Data loading."""
# pylint:disable=g-multiple-import

from jax_privacy.experiments.image_data.augmult import AugmultConfig
from jax_privacy.experiments.image_data.base import (
    DataInputs,
    DatasetConfig,
    ImageDatasetConfig,
)
from jax_privacy.experiments.image_data.imagenet import (
    ImageNetLoader,
    ImageNetConfig,
    ImagenetTestConfig,
    ImagenetTrainConfig,
    ImagenetTrainValidConfig,
    ImagenetValidConfig,
    ImageNetNumSamples,
)
from jax_privacy.experiments.image_data.loader import DataLoader
from jax_privacy.experiments.image_data.mnist_cifar_svhn import (
    MnistLoader,
    Cifar10Loader,
    Cifar100Loader,
    SvhnLoader,
    Cifar10TrainConfig,
    Cifar10TrainValidConfig,
    Cifar10ValidConfig,
    Cifar10TestConfig,
    Cifar100TrainConfig,
    Cifar100TrainValidConfig,
    Cifar100ValidConfig,
    Cifar100TestConfig,
    SvhnTrainConfig,
    SvhnValidConfig,
    SvhnTrainValidConfig,
    SvhnTestConfig,
    MnistTrainConfig,
    MnistValidConfig,
    MnistTrainValidConfig,
    MnistTestConfig,
)
from jax_privacy.experiments.image_data.places365 import (
    Places365Loader,
    Places365TrainConfig,
    Places365ValidConfig,
    Places365TrainValidConfig,
    Places365Testconfig,
    Places365NumSamples,
)
