"""Shared lightweight types for the frontend layer."""

from __future__ import annotations

from typing import NamedTuple

import numpy as np
import torch

ActionIdxBatch = list[list[int]]
TokenIds = list[int] | np.ndarray | torch.Tensor


class DecodeResult(NamedTuple):
    action: torch.Tensor
    absent_keys: set


__all__ = ["ActionIdxBatch", "TokenIds", "DecodeResult"]