[BugFix]: Fix local mypy issues (#34739)

Signed-off-by: Martin Hickey <martin.hickey@ie.ibm.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Martin Hickey
2026-02-23 08:40:29 +00:00
committed by GitHub
parent 7291d1b288
commit e97c46a92d
6 changed files with 12 additions and 6 deletions

View File

@@ -3,7 +3,7 @@
"""Sampling parameters for text generation."""
import copy
import json
import json as json_mod
from dataclasses import field
from enum import Enum, IntEnum
from functools import cached_property
@@ -791,7 +791,7 @@ class SamplingParams(
skip_guidance = False
if so_params.json:
if isinstance(so_params.json, str):
schema = json.loads(so_params.json)
schema = json_mod.loads(so_params.json)
else:
schema = so_params.json
skip_guidance = has_guidance_unsupported_json_features(schema)