Source code for pythae.models.normalizing_flows.base.base_nf_config

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