Update deprecated Python 3.8 typing (#13971)

This commit is contained in:
Harry Mellor
2025-03-03 01:34:51 +00:00
committed by GitHub
parent bf33700ecd
commit cf069aa8aa
300 changed files with 2294 additions and 2347 deletions

View File

@@ -1,7 +1,7 @@
# SPDX-License-Identifier: Apache-2.0
from dataclasses import dataclass
from typing import Dict, List, Optional, Set, Tuple
from typing import Optional
import torch
@@ -17,7 +17,7 @@ class SamplingMetadata:
top_k: Optional[torch.Tensor]
min_p: Optional[torch.Tensor]
generators: Dict[int, torch.Generator]
generators: dict[int, torch.Generator]
# None means no logprobs, 0 means sampled token logprobs only
max_num_logprobs: Optional[int]
@@ -28,12 +28,12 @@ class SamplingMetadata:
presence_penalties: torch.Tensor
repetition_penalties: torch.Tensor
output_token_ids: List[List[int]]
output_token_ids: list[list[int]]
# req_index -> (min_tokens, stop_token_ids)
min_tokens: Dict[int, Tuple[int, Set[int]]]
min_tokens: dict[int, tuple[int, set[int]]]
logit_bias: List[Optional[Dict[int, float]]]
logit_bias: list[Optional[dict[int, float]]]
# `allowed_token_ids_mask` is a 2D bool tensor of shape (max batch size,
# vocab size).