from typing import Tuple, Union

from pydantic.dataclasses import dataclass
from pythae.config import BaseConfig


@dataclass
class BaseNFConfig(BaseConfig):
    """This is the Base Normalizing Flow config instance.

    Parameters:
        input_dim (tuple): The input data dimension. Default: None.
    """

    input_dim: Union[Tuple[int, ...], None] = None

from typing import Tuple, Union

from pydantic.dataclasses import dataclass
from pythae.config import BaseConfig


@dataclass
class BaseNFConfig(BaseConfig):
    """This is the Base Normalizing Flow config instance.

    Parameters:
        input_dim (tuple): The input data dimension. Default: None.
    """

    input_dim: Union[Tuple[int, ...], None] = None

This is the Base Normalizing Flow config instance.      Parameters:         input_dim (tuple): The input data dimension. Default: None.