| """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"] | |
| """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"] | |