black and isort

landing-ai · Apr 25, 2024 · 3601d86 · 3601d86
1 parent 5e3c8cd
commit 3601d86
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 9 deletions.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -380,7 +380,6 @@ def sample_n_evenly_spaced(lst: Sequence, n: int) -> Sequence:
  return [lst[round(spacing * i)] for i in range(n)]
 
 
-
 def visualize_result(all_tool_results: List[Dict]) -> Sequence[Union[str, Path]]:
  image_to_data: Dict[str, Dict] = {}
  for tool_result in all_tool_results:

diff --git a/vision_agent/lmm/lmm.py b/vision_agent/lmm/lmm.py
@@ -9,10 +9,7 @@
 import requests
 from openai import AzureOpenAI, OpenAI
 
-from vision_agent.tools import (
- CHOOSE_PARAMS,
- SYSTEM_PROMPT,
-)
+from vision_agent.tools import CHOOSE_PARAMS, SYSTEM_PROMPT
 
 _LOGGER = logging.getLogger(__name__)
 

diff --git a/vision_agent/tools/__init__.py b/vision_agent/tools/__init__.py
@@ -12,12 +12,12 @@
  GroundingDINO,
  GroundingSAM,
  ImageCaption,
- ZeroShotCounting,
- VisualPromptCounting,
- VisualQuestionAnswering,
  ImageQuestionAnswering,
  SegArea,
  SegIoU,
  Tool,
+ VisualPromptCounting,
+ VisualQuestionAnswering,
+ ZeroShotCounting,
  register_tool,
 )
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -17,9 +17,9 @@
  normalize_bbox,
  rle_decode,
 )
+from vision_agent.lmm import OpenAILMM
 from vision_agent.tools.video import extract_frames_from_video
 from vision_agent.type_defs import LandingaiAPIKey
-from vision_agent.lmm import OpenAILMM
 
 _LOGGER = logging.getLogger(__name__)
 _LND_API_KEY = LandingaiAPIKey().api_key