phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-27 18:50:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: finish conversion to StrEnum (#2514)

Browse source 
# What does this PR do?

We still had a few enum declared to behave like string as well as enum.
Let's use StrEnum for those.

Signed-off-by: Sébastien Han <seb@redhat.com>
This commit is contained in:
Sébastien Han 2025-06-26 04:31:26 +02:00 • committed by GitHub
parent ac5fd57387
commit 36d70637b9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 17 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/models/llama/datatypes.py
View file

@ -5,7 +5,7 @@
# the root directory of this source tree.
import base64
from enum import Enum
from enum import Enum, StrEnum
from io import BytesIO
from typing import Annotated, Any, Literal
@ -171,7 +171,7 @@ class GenerationResult(BaseModel):
ignore_token: bool
class QuantizationMode(str, Enum):
class QuantizationMode(StrEnum):
none = "none"
fp8_mixed = "fp8_mixed"
int4_mixed = "int4_mixed"