Skip to content

Commit 91651b0

Browse files
committed
Ignore issue with pydantic typing + other fixes
1 parent 26bb3a3 commit 91651b0

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/guidellm/benchmark/scenario.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
from pathlib import Path
2-
from typing import Any, Dict, Literal, Optional, Self, Union
2+
from typing import Any, Dict, Literal, Optional, Union
33

44
from pydantic import Field
5+
from typing_extensions import Self
56

67
from guidellm.backend import BackendType
78
from guidellm.core import Serializable
@@ -17,7 +18,7 @@ class Scenario(Serializable):
1718
backend_kwargs: Optional[Dict[str, Any]] = None
1819
model: Optional[str] = None
1920
tokenizer: Optional[str] = None
20-
data: Union[str, Dict[str, Any]] = Field(default_factory=dict)
21+
data: Union[str, Dict[str, Any]] = Field(default_factory=dict) # type: ignore[arg-type]
2122
data_type: Literal["emulated", "file", "transformers"] = "emulated"
2223
rate_type: ProfileGenerationMode = "sweep"
2324
rate: Optional[float] = None

src/guidellm/utils/cli_params.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def __init__(self, *types: ParamType):
4343
self.types = types
4444
self.name = "".join(t.name for t in types)
4545

46-
def convert(self, value, param, ctx):
46+
def convert(self, value, param, ctx) -> Any:
4747
fails = []
4848
for t in self.types:
4949
try:

0 commit comments

Comments
 (0)