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

from __future__ import annotations

import numpy as np
import torch

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


__all__ = ["ActionIdxBatch", "TokenIds"]