From 1c4fccae17bb043b3e4b196669dc0777524037fc Mon Sep 17 00:00:00 2001
From: Bagatur <baskaryan@gmail.com>
Date: Thu, 31 Oct 2024 12:09:45 -0700
Subject: [PATCH] fmt

---
 libs/vertexai/tests/integration_tests/conftest.py         | 4 ----
 libs/vertexai/tests/integration_tests/test_chat_models.py | 1 -
 2 files changed, 5 deletions(-)

diff --git a/libs/vertexai/tests/integration_tests/conftest.py b/libs/vertexai/tests/integration_tests/conftest.py
index 6277948e..7fe726da 100644
--- a/libs/vertexai/tests/integration_tests/conftest.py
+++ b/libs/vertexai/tests/integration_tests/conftest.py
@@ -1,8 +1,4 @@
-import base64
-
 import pytest
-from _pytest.tmpdir import TempPathFactory
-from vertexai.vision_models import Image  # type: ignore
 
 _DEFAULT_MODEL_NAME = "gemini-1.0-pro-001"
 
diff --git a/libs/vertexai/tests/integration_tests/test_chat_models.py b/libs/vertexai/tests/integration_tests/test_chat_models.py
index 4945aec7..ba332003 100644
--- a/libs/vertexai/tests/integration_tests/test_chat_models.py
+++ b/libs/vertexai/tests/integration_tests/test_chat_models.py
@@ -4,7 +4,6 @@
 import io
 import json
 import os
-import tempfile
 from typing import List, Literal, Optional, cast
 
 import pytest