File size: 377 Bytes
7c09a0c 1bfc3f0 7c09a0c 1bfc3f0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | """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"]
|