From 1a18c7d7f8074bc9cff93b45d040f3e52250fac5 Mon Sep 17 00:00:00 2001 From: Teddy Date: Tue, 11 Mar 2025 15:52:39 -0700 Subject: [PATCH] fix: default to raw dataset (#20192) --- ingestion/src/metadata/sampler/sqlalchemy/sampler.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ingestion/src/metadata/sampler/sqlalchemy/sampler.py b/ingestion/src/metadata/sampler/sqlalchemy/sampler.py index 4d28f29f4a9..7a44cd44f77 100644 --- a/ingestion/src/metadata/sampler/sqlalchemy/sampler.py +++ b/ingestion/src/metadata/sampler/sqlalchemy/sampler.py @@ -153,7 +153,10 @@ class SQASampler(SamplerInterface, SQAInterfaceMixin): if self.sample_query: return self._rdn_sample_from_user_query() - if not self.sample_config.profileSample: + if not self.sample_config.profileSample or ( + self.sample_config.profileSampleType == ProfileSampleType.PERCENTAGE + and self.sample_config.profileSample == 100 + ): if self.partition_details: partitioned = self._partitioned_table() return partitioned.cte(f"{self.get_sampler_table_name()}_partitioned")