From 482b3ccf63621b0b3d1c8d54d6d7d27312af409d Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Mon, 11 Mar 2024 19:38:44 -0700 Subject: [PATCH] isort --- vision_agent/__init__.py | 6 +++--- vision_agent/data/data.py | 2 +- vision_agent/image_utils.py | 2 +- vision_agent/llm/llm.py | 2 +- vision_agent/tools/__init__.py | 4 ++-- 5 files changed, 8 insertions(+), 8 deletions(-) diff --git a/vision_agent/__init__.py b/vision_agent/__init__.py index ccb910f8..8d9c25b0 100644 --- a/vision_agent/__init__.py +++ b/vision_agent/__init__.py @@ -1,4 +1,4 @@ -from .lmm import LMM, LLaVALMM, OpenAILMM, get_lmm -from .llm import LLM, OpenAILLM -from .emb import Embedder, SentenceTransformerEmb, OpenAIEmb, get_embedder from .data import DataStore, build_data_store +from .emb import Embedder, OpenAIEmb, SentenceTransformerEmb, get_embedder +from .llm import LLM, OpenAILLM +from .lmm import LMM, LLaVALMM, OpenAILMM, get_lmm diff --git a/vision_agent/data/data.py b/vision_agent/data/data.py index bdf381a9..4548b42b 100644 --- a/vision_agent/data/data.py +++ b/vision_agent/data/data.py @@ -2,7 +2,7 @@ import uuid from pathlib import Path -from typing import Dict, List, Optional, Union, cast, Callable +from typing import Callable, Dict, List, Optional, Union, cast import faiss import numpy as np diff --git a/vision_agent/image_utils.py b/vision_agent/image_utils.py index c99b213e..c48d5620 100644 --- a/vision_agent/image_utils.py +++ b/vision_agent/image_utils.py @@ -1,7 +1,7 @@ import base64 from io import BytesIO from pathlib import Path -from typing import Union, Tuple +from typing import Tuple, Union import numpy as np from PIL import Image diff --git a/vision_agent/llm/llm.py b/vision_agent/llm/llm.py index 587d4b28..6931ca03 100644 --- a/vision_agent/llm/llm.py +++ b/vision_agent/llm/llm.py @@ -1,6 +1,6 @@ import json -from typing import cast from abc import ABC, abstractmethod +from typing import cast from vision_agent.tools import ( CHOOSE_PARAMS, diff --git a/vision_agent/tools/__init__.py b/vision_agent/tools/__init__.py index eedfb21d..2f7fe9be 100644 --- a/vision_agent/tools/__init__.py +++ b/vision_agent/tools/__init__.py @@ -1,2 +1,2 @@ -from .prompts import SYSTEM_PROMPT, CHOOSE_PARAMS -from .tools import ImageTool, CLIP, GroundingDINO, GroundingSAM +from .prompts import CHOOSE_PARAMS, SYSTEM_PROMPT +from .tools import CLIP, GroundingDINO, GroundingSAM, ImageTool