@@ -87,8 +87,6 @@ def assert_client_called_once_with(
8787
8888
8989# --- FIXTURES ---
90-
91-
9290@pytest .fixture
9391def mock_dataset_service_client_class ():
9492 with mock .patch (
@@ -115,9 +113,10 @@ def mock_model_service_client_class():
115113 ) as mock_class :
116114 yield mock_class
117115
118-
119116# --- TEST CLASSES ---
120117
118+ from google .api_core import client_options as client_options_lib
119+ from google .auth import credentials as auth_credentials
121120
122121class TestCentralizedClientInitialization :
123122 @pytest .mark .parametrize (
@@ -249,14 +248,6 @@ class TestCentralizedClientModelService:
249248 def test_get_model (self , mock_model_service_client_class ):
250249 # Arrange
251250 mock_instance = mock_model_service_client_class .return_value
252- expected_model = model .Model (
253- etag = DEFAULT_ETAG ,
254- model_reference = {
255- "project_id" : PROJECT_ID ,
256- "dataset_id" : DATASET_ID ,
257- "model_id" : MODEL_ID ,
258- },
259- )
260251 mock_instance .get_model .return_value = expected_model
261252 get_model_request = model .GetModelRequest (
262253 project_id = PROJECT_ID , dataset_id = DATASET_ID , model_id = MODEL_ID
@@ -299,7 +290,6 @@ def test_patch_model(self, mock_model_service_client_class):
299290 description = "A newly patched description." ,
300291 )
301292 mock_instance .patch_model .return_value = expected_model
302-
303293 model_patch = model .Model (description = "A newly patched description." )
304294 patch_model_request = model .PatchModelRequest (
305295 project_id = PROJECT_ID ,
0 commit comments