# SPDX-FileCopyrightText: 2024-2025 ModelCloud.ai
# SPDX-FileCopyrightText: 2024-2025 qubitium@modelcloud.ai
# SPDX-License-Identifier: Apache-2.0
# Contact: qubitium@modelcloud.ai, x.com/qubitium

from dataclasses import dataclass
from enum import Enum


class LinearMode(str, Enum):
    """Selects whether wrapped linear modules run inference or training behavior."""

    INFERENCE = "inference"
    TRAIN = "train"