import torch
from torchaudio_unittest.common_utils import PytorchTestCase

from .model_test_impl import Tacotron2DecoderTests, Tacotron2EncoderTests, Tacotron2Tests


class TestTacotron2EncoderFloat32CPU(Tacotron2EncoderTests, PytorchTestCase):
    dtype = torch.float32
    device = torch.device("cpu")


class TestTacotron2DecoderFloat32CPU(Tacotron2DecoderTests, PytorchTestCase):
    dtype = torch.float32
    device = torch.device("cpu")


class TestTacotron2Float32CPU(Tacotron2Tests, PytorchTestCase):
    dtype = torch.float32
    device = torch.device("cpu")

import torch
from torchaudio_unittest.common_utils import PytorchTestCase

from .model_test_impl import (
    Tacotron2DecoderTests,
    Tacotron2EncoderTests,
    Tacotron2Tests,
)


class TestTacotron2EncoderFloat32CPU(Tacotron2EncoderTests, PytorchTestCase):
    dtype = torch.float32
    device = torch.device("cpu")


class TestTacotron2DecoderFloat32CPU(Tacotron2DecoderTests, PytorchTestCase):
    dtype = torch.float32
    device = torch.device("cpu")


class TestTacotron2Float32CPU(Tacotron2Tests, PytorchTestCase):
    dtype = torch.float32
    device = torch.device("cpu")
