Skip to content

Commit

Permalink
fix linting error
Browse files Browse the repository at this point in the history
  • Loading branch information
dillonalaird committed Aug 26, 2024
1 parent a75b892 commit 25154c2
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 18 deletions.
5 changes: 0 additions & 5 deletions vision_agent/tools/meta_tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,16 +2,11 @@
import subprocess
from pathlib import Path
from typing import Any, Dict, List, Union
from uuid import UUID

import vision_agent as va
from vision_agent.clients.landing_public_api import LandingPublicAPI
from vision_agent.lmm.types import Message
from vision_agent.tools.meta_tools_types import BboxInput, BboxInputBase64, PromptTask
from vision_agent.tools.tool_utils import get_tool_documentation
from vision_agent.tools.tools import TOOL_DESCRIPTIONS
from vision_agent.utils.image_utils import convert_to_b64


# These tools are adapted from SWE-Agent https://github.com/princeton-nlp/SWE-agent

Expand Down
25 changes: 12 additions & 13 deletions vision_agent/tools/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,10 @@
import json
import logging
import tempfile
from uuid import UUID
from pathlib import Path
from importlib import resources
from pathlib import Path
from typing import Any, Dict, List, Optional, Tuple, Union, cast
from uuid import UUID

import cv2
import numpy as np
Expand All @@ -16,15 +15,24 @@
from pillow_heif import register_heif_opener # type: ignore
from pytube import YouTube # type: ignore

from vision_agent.clients.landing_public_api import LandingPublicAPI
from vision_agent.tools.tool_utils import (
get_tool_descriptions,
get_tool_documentation,
get_tools_df,
send_inference_request,
get_tools_info,
send_inference_request,
)
from vision_agent.tools.tools_types import (
BboxInput,
BboxInputBase64,
FineTuning,
Florencev2FtRequest,
JobStatus,
PromptTask,
)
from vision_agent.utils.exceptions import FineTuneModelIsNotReady
from vision_agent.utils import extract_frames_from_video
from vision_agent.utils.exceptions import FineTuneModelIsNotReady
from vision_agent.utils.execute import FileSerializer, MimeType
from vision_agent.utils.image_utils import (
b64_to_pil,
Expand All @@ -38,15 +46,6 @@
rle_decode,
rle_decode_array,
)
from vision_agent.tools.tools_types import (
BboxInput,
BboxInputBase64,
PromptTask,
Florencev2FtRequest,
FineTuning,
JobStatus,
)
from vision_agent.clients.landing_public_api import LandingPublicAPI

register_heif_opener()

Expand Down

0 comments on commit 25154c2

Please sign in to comment.