Skip to content

Commit 4a528c6

Browse files
jaycee-licopybara-github
authored andcommitted
chore: GenAI - Removed unused tensorflow and pyfakefs dependencies
PiperOrigin-RevId: 670425612
1 parent 8015be3 commit 4a528c6

File tree

2 files changed

+0
-17
lines changed

2 files changed

+0
-17
lines changed

setup.py

-1
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,6 @@
201201
"grpcio-testing",
202202
"ipython",
203203
"kfp >= 2.6.0, < 3.0.0",
204-
"pyfakefs",
205204
"pytest-asyncio",
206205
"pytest-xdist",
207206
"scikit-learn",

tests/unit/vertexai/conftest.py

-16
Original file line numberDiff line numberDiff line change
@@ -65,9 +65,7 @@
6565
_TEST_OPTIMIZED_FV2,
6666
_TEST_PSC_OPTIMIZED_FOS,
6767
)
68-
from pyfakefs import fake_filesystem_unittest
6968
import pytest
70-
import tensorflow.saved_model as tf_saved_model
7169

7270
_TEST_PROJECT = "test-project"
7371
_TEST_PROJECT_NUMBER = "12345678"
@@ -141,14 +139,6 @@ def google_auth_mock():
141139
yield auth_mock
142140

143141

144-
@pytest.fixture
145-
def mock_filesystem():
146-
with fake_filesystem_unittest.Patcher() as patcher:
147-
patcher.setUp()
148-
yield patcher.fs
149-
patcher.tearDown()
150-
151-
152142
@pytest.fixture
153143
def mock_storage_blob(mock_filesystem):
154144
"""Mocks the storage Blob API.
@@ -291,12 +281,6 @@ def mock_uuid():
291281
yield uuid_mock
292282

293283

294-
@pytest.fixture
295-
def mock_tf_saved_model_load():
296-
with mock.patch.object(tf_saved_model, "load") as load_mock:
297-
yield load_mock
298-
299-
300284
@pytest.fixture
301285
def base_logger_mock():
302286
with patch.object(

0 commit comments

Comments
 (0)