Source code for marivo.analysis.intents._types

"""Shared public type aliases for analysis intents."""

from __future__ import annotations

from typing import Any, Literal, TypedDict

SlicePredicateOp = Literal["==", "!=", "in", ">", ">=", "<", "<=", "between"]




class SlicePredicate(TypedDict):
    op: SlicePredicateOp
    value: Any



SliceScalar = str | int | float | bool | None
SliceValue = SliceScalar | list[SliceScalar] | SlicePredicate

DiscoverSensitivity = Literal["conservative", "balanced", "aggressive"]

__all__ = [
    "DiscoverSensitivity",
    "SlicePredicate",
    "SlicePredicateOp",
    "SliceScalar",
    "SliceValue",
]