Cleaned up mocks

samples/model-builder/conftest.py

@@ -215,7 +215,7 @@ def mock_run_custom_training_job():
 
 @pytest.fixture
 def mock_model():
-    mock = MagicMock(aiplatform.models.Model)
+    mock = MagicMock(aiplatform.Model)
     yield mock
 
 
@@ -231,11 +231,13 @@ def mock_batch_predict_model(mock_model):
     with patch.object(mock_model, "batch_predict") as mock:
         yield mock
 
+
 @pytest.fixture
 def mock_upload_model():
     with patch.object(aiplatform.Model, "upload") as mock:
         yield mock
 
+
 @pytest.fixture
 def mock_deploy_model(mock_model, mock_endpoint):
     with patch.object(mock_model, "deploy") as mock:
@@ -265,7 +267,7 @@ def mock_create_batch_prediction_job():
 
 @pytest.fixture
 def mock_endpoint():
-    mock = MagicMock(aiplatform.models.Endpoint)
+    mock = MagicMock(aiplatform.Endpoint)
     yield mock
 
 
@@ -281,8 +283,9 @@ def mock_get_endpoint(mock_endpoint):
         mock_get_endpoint.return_value = mock_endpoint
         yield mock_get_endpoint
 
+
 @pytest.fixture
 def mock_endpoint_explain(mock_endpoint):
     with patch.object(mock_endpoint, "explain") as mock_endpoint_explain:
         mock_get_endpoint.return_value = mock_endpoint
-        yield mock_endpoint_explain
+        yield mock_endpoint_explain