diff --git a/ingestion/tests/cli_e2e/base/config_builders/builders.py b/ingestion/tests/cli_e2e/base/config_builders/builders.py index 7d4824f27bcd..f8baf2b0dc26 100644 --- a/ingestion/tests/cli_e2e/base/config_builders/builders.py +++ b/ingestion/tests/cli_e2e/base/config_builders/builders.py @@ -58,7 +58,12 @@ def build(self) -> dict: self.config["source"]["sourceConfig"] = { "config": { "type": "Profiler", - "profileSample": self.profilerSample, + "profileSampleConfig": { + "sampleConfigType": "STATIC", + "config": { + "profileSample": self.profilerSample, + }, + }, } } diff --git a/ingestion/tests/cli_e2e/test_cli_bigquery.py b/ingestion/tests/cli_e2e/test_cli_bigquery.py index f6e120ee80eb..29bef2b8b420 100644 --- a/ingestion/tests/cli_e2e/test_cli_bigquery.py +++ b/ingestion/tests/cli_e2e/test_cli_bigquery.py @@ -30,6 +30,11 @@ from metadata.generated.schema.tests.basic import TestCaseResult, TestCaseStatus from metadata.generated.schema.tests.testCase import TestCaseParameterValue from metadata.generated.schema.type.basic import ProfileSampleType, Timestamp +from metadata.generated.schema.type.samplingConfig import ( + ProfileSampleConfig, + SampleConfigType, +) +from metadata.generated.schema.type.staticSamplingConfig import StaticSamplingConfig from .common.test_cli_db import CliCommonDB # noqa: TID252 from .common_e2e_sqa_mixins import SQACommonMethods # noqa: TID252 @@ -189,8 +194,13 @@ def add_table_profile_config(self): self.openmetadata.create_or_update_table_profiler_config( self.get_data_quality_table(), TableProfilerConfig( - profileSampleType=ProfileSampleType.ROWS, - profileSample=100, + profileSampleConfig=ProfileSampleConfig( + sampleConfigType=SampleConfigType.STATIC, + config=StaticSamplingConfig( + profileSample=100, + profileSampleType=ProfileSampleType.ROWS, + ), + ), ), )