fix(azure_ai): specify FLUX parameter mapping return type

This commit is contained in:
Emerson Gomes 2026-09-15 12:43:24 -05:00
parent ff1e2a02ba
commit 3821e5ace6
No known key found for this signature in database
GPG key ID: D3DF28AB5D1B5E17

View file

@ -102,7 +102,7 @@ class AzureFoundryFluxImageGenerationConfig(GPTImageGenerationConfig):
optional_params: Mapping[str, object],
model: str,
drop_params: bool,
) -> dict: # mutable-ok: inherited config contract returns a dict
) -> dict[str, object]: # mutable-ok: inherited config contract returns a dict
if not self.is_flux2_model(model):
return super().map_openai_params(
non_default_params=dict(non_default_params),