Skip to content

Commit 35f0081

Browse files
committed
renamed mock data file
1 parent 55e9cfc commit 35f0081

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

metadata-ingestion/tests/integration/vertexai/test_vertexai.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from pytest import Config
88

99
from datahub.ingestion.run.pipeline import Pipeline
10-
from tests.integration.vertexai.mock_vertexai_entity import (
10+
from tests.integration.vertexai.mock_vertexai import (
1111
gen_mock_dataset,
1212
gen_mock_model,
1313
gen_mock_models,
@@ -68,7 +68,6 @@ def test_vertexai_source_ingestion(pytestconfig: Config, sink_file_path: str) ->
6868
mock = exit_stack.enter_context(patch(func_to_mock))
6969
mock.return_value = []
7070

71-
7271
# Mock the Vertex AI with Mock data list
7372
mock_models = exit_stack.enter_context(
7473
patch("google.cloud.aiplatform.Model.list")

metadata-ingestion/tests/unit/test_vertexai_source.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
MLTrainingRunPropertiesClass,
3030
SubTypesClass,
3131
)
32-
from tests.integration.vertexai.mock_vertexai_entity import (
32+
from tests.integration.vertexai.mock_vertexai import (
3333
gen_mock_dataset,
3434
gen_mock_endpoint,
3535
gen_mock_model,

0 commit comments

Comments
 (0)