phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-29 03:14:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix precommit check after moving to ruff (#927)

Browse source 
Lint check in main branch is failing. This fixes the lint check after we
moved to ruff in https://github.com/meta-llama/llama-stack/pull/921. We
need to move to a `ruff.toml` file as well as fixing and ignoring some
additional checks.

Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2025-02-02 09:46:45 -05:00 • committed by GitHub
parent 4773092dd1
commit 34ab7a3b6c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
217 changed files with 981 additions and 2681 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/templates/template.py
View file

@ -53,9 +53,7 @@ class RunConfigSettings(BaseModel):
api = Api(api_str)
if provider_type not in provider_registry[api]:
raise ValueError(
f"Unknown provider type: {provider_type} for API: {api_str}"
)
raise ValueError(f"Unknown provider type: {provider_type} for API: {api_str}")
config_class = provider_registry[api][provider_type].config_class
assert config_class is not None, (
@ -64,9 +62,7 @@ class RunConfigSettings(BaseModel):
config_class = instantiate_class_type(config_class)
if hasattr(config_class, "sample_run_config"):
config = config_class.sample_run_config(
__distro_dir__=f"distributions/{name}"
)
config = config_class.sample_run_config(__distro_dir__=f"distributions/{name}")
else:
config = {}
@ -79,7 +75,7 @@ class RunConfigSettings(BaseModel):
)
# Get unique set of APIs from providers
apis = list(sorted(providers.keys()))
apis = sorted(providers.keys())
return StackRunConfig(
image_name=name,
@ -173,9 +169,7 @@ class DistributionTemplate(BaseModel):
)
for yaml_pth, settings in self.run_configs.items():
run_config = settings.run_config(
self.name, self.providers, self.container_image
)
run_config = settings.run_config(self.name, self.providers, self.container_image)
with open(yaml_output_dir / yaml_pth, "w") as f:
yaml.safe_dump(
run_config.model_dump(exclude_none=True),