pre-commit lint

yanxi0830 · yanxi0830 · commit 4ae8c63a2b4d · 2024-09-28T16:04:41.000-07:00
diff --git a/llama_stack/apis/inference/event_logger.py b/llama_stack/apis/inference/event_logger.py
@@ -4,11 +4,12 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
+from termcolor import cprint
+
 from llama_stack.apis.inference import (
     ChatCompletionResponseEventType,
     ChatCompletionResponseStreamChunk,
 )
-from termcolor import cprint
 
 
 class LogEvent:
diff --git a/llama_stack/cli/model/describe.py b/llama_stack/cli/model/describe.py
@@ -9,12 +9,12 @@
 
 from llama_models.sku_list import resolve_model
 
+from termcolor import colored
+
 from llama_stack.cli.subcommand import Subcommand
 from llama_stack.cli.table import print_table
 from llama_stack.distribution.utils.serialize import EnumEncoder
 
-from termcolor import colored
-
 
 class ModelDescribe(Subcommand):
     """Show details about a model"""
diff --git a/llama_stack/cli/stack/run.py b/llama_stack/cli/stack/run.py
@@ -46,6 +46,7 @@ def _run_stack_run_cmd(self, args: argparse.Namespace) -> None:
 
         import pkg_resources
         import yaml
+
         from llama_stack.distribution.build import ImageType
         from llama_stack.distribution.utils.config_dirs import BUILDS_BASE_DIR
 
diff --git a/llama_stack/distribution/configure.py b/llama_stack/distribution/configure.py
@@ -9,6 +9,10 @@
 from pydantic import BaseModel
 
 from llama_stack.distribution.datatypes import *  # noqa: F403
+from prompt_toolkit import prompt
+from prompt_toolkit.validation import Validator
+from termcolor import cprint
+
 from llama_stack.apis.memory.memory import MemoryBankType
 from llama_stack.distribution.distribution import (
     api_providers,
@@ -21,9 +25,6 @@
 from llama_stack.providers.impls.meta_reference.safety.config import (
     MetaReferenceShieldType,
 )
-from prompt_toolkit import prompt
-from prompt_toolkit.validation import Validator
-from termcolor import cprint
 
 
 def make_routing_entry_type(config_class: Any):
diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py
@@ -435,13 +435,13 @@ async def healthcheck():
         apis_to_serve = set(config.apis_to_serve)
     else:
         apis_to_serve = set(impls.keys())
-    
+
     for api_str in apis_to_serve:
         api = Api(api_str)
 
         endpoints = all_endpoints[api]
         impl = impls[api]
-        
+
         provider_spec = specs[api]
         if (
             isinstance(provider_spec, RemoteProviderSpec)
diff --git a/llama_stack/distribution/utils/config_dirs.py b/llama_stack/distribution/utils/config_dirs.py
@@ -8,7 +8,9 @@
 from pathlib import Path
 
 
-LLAMA_STACK_CONFIG_DIR = Path(os.getenv("LLAMA_STACK_CONFIG_DIR", os.path.expanduser("~/.llama/")))
+LLAMA_STACK_CONFIG_DIR = Path(
+    os.getenv("LLAMA_STACK_CONFIG_DIR", os.path.expanduser("~/.llama/"))
+)
 
 DISTRIBS_BASE_DIR = LLAMA_STACK_CONFIG_DIR / "distributions"
 
diff --git a/llama_stack/distribution/utils/dynamic.py b/llama_stack/distribution/utils/dynamic.py
@@ -8,7 +8,6 @@
 from typing import Any, Dict
 
 from llama_stack.distribution.datatypes import *  # noqa: F403
-from termcolor import cprint
 
 
 def instantiate_class_type(fully_qualified_name):
diff --git a/llama_stack/providers/adapters/inference/together/together.py b/llama_stack/providers/adapters/inference/together/together.py
@@ -15,10 +15,10 @@
 from together import Together
 
 from llama_stack.apis.inference import *  # noqa: F403
+from llama_stack.distribution.request_headers import get_request_provider_data
 from llama_stack.providers.utils.inference.augment_messages import (
     augment_messages_for_tools,
 )
-from llama_stack.distribution.request_headers import get_request_provider_data
 
 from .config import TogetherImplConfig
 
diff --git a/llama_stack/providers/impls/meta_reference/inference/quantization/loader.py b/llama_stack/providers/impls/meta_reference/inference/quantization/loader.py
@@ -14,16 +14,17 @@
 
 from fairscale.nn.model_parallel.mappings import reduce_from_model_parallel_region
 from llama_models.llama3.api.model import Transformer, TransformerBlock
+
+from termcolor import cprint
+from torch import Tensor
+
 from llama_stack.apis.inference import QuantizationType
 
 from llama_stack.apis.inference.config import (
     CheckpointQuantizationFormat,
     MetaReferenceImplConfig,
 )
 
-from termcolor import cprint
-from torch import Tensor
-
 
 def is_fbgemm_available() -> bool:
     try:

Original file line number	Diff line number	Diff line change
`@@ -15,10 +15,10 @@`
`15`	`15`	`from together import Together`
`16`	`16`
`17`	`17`	`from llama_stack.apis.inference import * # noqa: F403`
	`18`	`+from llama_stack.distribution.request_headers import get_request_provider_data`
`18`	`19`	`from llama_stack.providers.utils.inference.augment_messages import (`
`19`	`20`	`augment_messages_for_tools,`
`20`	`21`	`)`
`21`		`-from llama_stack.distribution.request_headers import get_request_provider_data`
`22`	`22`
`23`	`23`	`from .config import TogetherImplConfig`
`24`	`24`