phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-22 22:42:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: add mypy llama4 files

Browse source 
Signed-off-by: Mustafa Elbehery <melbeher@redhat.com>
This commit is contained in:
Mustafa Elbehery 2025-07-08 23:51:33 +02:00
parent 51b179e1c5
commit eb988a0d4e
2 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/models/llama/llama4/tokenizer.py
View file

@ -237,7 +237,7 @@ class Tokenizer:
str: The decoded string. str: The decoded string.
""" """
# Typecast is safe here. Tiktoken doesn't do anything list-related with the sequence. # Typecast is safe here. Tiktoken doesn't do anything list-related with the sequence.
return self.model.decode(cast(list[int], t)) return cast(str, self.model.decode(cast(list[int], t)))
@staticmethod @staticmethod
def _split_whitespaces_or_nonwhitespaces(s: str, max_consecutive_slice_len: int) -> Iterator[str]: def _split_whitespaces_or_nonwhitespaces(s: str, max_consecutive_slice_len: int) -> Iterator[str]:

14
pyproject.toml
View file

@ -254,7 +254,19 @@ exclude = [
"^llama_stack/providers/inline/inference/meta_reference/inference\\.py$", "^llama_stack/providers/inline/inference/meta_reference/inference\\.py$",
"^llama_stack/models/llama/llama3/generation\\.py$", "^llama_stack/models/llama/llama3/generation\\.py$",
"^llama_stack/models/llama/llama3/multimodal/model\\.py$", "^llama_stack/models/llama/llama3/multimodal/model\\.py$",
"^llama_stack/models/llama/llama4/", "^llama_stack/models/llama/llama4/args\\.py$",
"^llama_stack/models/llama/llama4/chat_format\\.py$",
"^llama_stack/models/llama/llama4/datatypes\\.py$",
"^llama_stack/models/llama/llama4/ffn\\.py$",
"^llama_stack/models/llama/llama4/generation\\.py$",
"^llama_stack/models/llama/llama4/model\\.py$",
"^llama_stack/models/llama/llama4/moe\\.py$",
"^llama_stack/models/llama/llama4/preprocess\\.py$",
"^llama_stack/models/llama/llama4/prompts\\.py$",
"^llama_stack/models/llama/llama4/prompt_templates/system_prompts\\.py$",
"^llama_stack/models/llama/llama4/quantization/loader\\.py$",
"^llama_stack/models/llama/llama4/vision/embedding\\.py$",
"^llama_stack/models/llama/llama4/vision/encoder\\.py$",
"^llama_stack/providers/inline/inference/meta_reference/parallel_utils\\.py$", "^llama_stack/providers/inline/inference/meta_reference/parallel_utils\\.py$",
"^llama_stack/providers/inline/inference/meta_reference/quantization/fp8_impls\\.py$", "^llama_stack/providers/inline/inference/meta_reference/quantization/fp8_impls\\.py$",
"^llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers\\.py$", "^llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers\\.py$",