Convert formatting to use ruff instead of yapf + isort (#26247)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -54,19 +54,22 @@ class MultiModelKeys(ModelKeys):
|
||||
generator: list[str] = field(default_factory=list)
|
||||
|
||||
@staticmethod
|
||||
def from_string_field(language_model: Union[str, list[str]] = None,
|
||||
connector: Union[str, list[str]] = None,
|
||||
tower_model: Union[str, list[str]] = None,
|
||||
generator: Union[str, list[str]] = None,
|
||||
**kwargs) -> 'MultiModelKeys':
|
||||
|
||||
def from_string_field(
|
||||
language_model: Union[str, list[str]] = None,
|
||||
connector: Union[str, list[str]] = None,
|
||||
tower_model: Union[str, list[str]] = None,
|
||||
generator: Union[str, list[str]] = None,
|
||||
**kwargs,
|
||||
) -> "MultiModelKeys":
|
||||
def to_list(value):
|
||||
if value is None:
|
||||
return []
|
||||
return [value] if isinstance(value, str) else list(value)
|
||||
|
||||
return MultiModelKeys(language_model=to_list(language_model),
|
||||
connector=to_list(connector),
|
||||
tower_model=to_list(tower_model),
|
||||
generator=to_list(generator),
|
||||
**kwargs)
|
||||
return MultiModelKeys(
|
||||
language_model=to_list(language_model),
|
||||
connector=to_list(connector),
|
||||
tower_model=to_list(tower_model),
|
||||
generator=to_list(generator),
|
||||
**kwargs,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user