renamed mock data file

ryota-cloud · ryota-cloud · commit 35f0081c8a3a · 2025-03-09T14:37:59.000-07:00
diff --git a/metadata-ingestion/tests/integration/vertexai/test_vertexai.py b/metadata-ingestion/tests/integration/vertexai/test_vertexai.py
@@ -7,7 +7,7 @@
 from pytest import Config
 
 from datahub.ingestion.run.pipeline import Pipeline
-from tests.integration.vertexai.mock_vertexai_entity import (
+from tests.integration.vertexai.mock_vertexai import (
     gen_mock_dataset,
     gen_mock_model,
     gen_mock_models,
@@ -68,7 +68,6 @@ def test_vertexai_source_ingestion(pytestconfig: Config, sink_file_path: str) ->
             mock = exit_stack.enter_context(patch(func_to_mock))
             mock.return_value = []
 
-
         # Mock the Vertex AI with Mock data list
         mock_models = exit_stack.enter_context(
             patch("google.cloud.aiplatform.Model.list")
diff --git a/metadata-ingestion/tests/unit/test_vertexai_source.py b/metadata-ingestion/tests/unit/test_vertexai_source.py
@@ -29,7 +29,7 @@
     MLTrainingRunPropertiesClass,
     SubTypesClass,
 )
-from tests.integration.vertexai.mock_vertexai_entity import (
+from tests.integration.vertexai.mock_vertexai import (
     gen_mock_dataset,
     gen_mock_endpoint,
     gen_mock_model,

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@`
`29`	`29`	`MLTrainingRunPropertiesClass,`
`30`	`30`	`SubTypesClass,`
`31`	`31`	`)`
`32`		`-from tests.integration.vertexai.mock_vertexai_entity import (`
	`32`	`+from tests.integration.vertexai.mock_vertexai import (`
`33`	`33`	`gen_mock_dataset,`
`34`	`34`	`gen_mock_endpoint,`
`35`	`35`	`gen_mock_model,`