Source code for mmagic.utils.typing

# Copyright (c) OpenMMLab. All rights reserved.
from typing import Callable, Dict, List, Sequence, Tuple, Union

from mmengine.config import ConfigDict
from mmengine.structures import BaseDataElement
from torch import Tensor

[docs]ForwardInputs = Tuple[Dict[str, Union[Tensor, str, int]], Tensor]
[docs]SampleList = Sequence[BaseDataElement]

[docs]NoiseVar = Union[Tensor, Callable, None]
[docs]LabelVar = Union[Tensor, Callable, List[int], None]

[docs]ConfigType = Union[ConfigDict, Dict]