diff --git a/llama_stack/providers/remote/inference/databricks/databricks.py b/llama_stack/providers/remote/inference/databricks/databricks.py
index b430dd411..25fd9f3b7 100644
--- a/llama_stack/providers/remote/inference/databricks/databricks.py
+++ b/llama_stack/providers/remote/inference/databricks/databricks.py
@@ -23,6 +23,8 @@ from llama_stack.apis.inference import (
     Inference,
     LogProbConfig,
     Message,
+    Model,
+    ModelType,
     OpenAICompletion,
     ResponseFormat,
     SamplingParams,
@@ -31,15 +33,12 @@ from llama_stack.apis.inference import (
     ToolConfig,
     ToolDefinition,
     ToolPromptFormat,
-    Model,
-    ModelType,
 )
+from llama_stack.log import get_logger
 from llama_stack.providers.utils.inference.openai_mixin import OpenAIMixin
 
 from .config import DatabricksImplConfig
 
-from llama_stack.log import get_logger
-
 logger = get_logger(name=__name__, category="inference::databricks")