minor change

datahub-project · Jul 16, 2024 · 600f515 · 600f515
1 parent 7010f3c
commit 600f515
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/bigquery_v2/bigquery_config.py b/metadata-ingestion/src/datahub/ingestion/source/bigquery_v2/bigquery_config.py
@@ -24,7 +24,9 @@
 
 logger = logging.getLogger(__name__)
 
-SCHEMA_PARALLELISM = int(os.getenv("DATAHUB_BIGQUERY_SCHEMA_PARALLELISM", 20))
+DEFAULT_BQ_SCHEMA_PARALLELISM = int(
+    os.getenv("DATAHUB_BIGQUERY_SCHEMA_PARALLELISM", 20)
+)
 
 
 class BigQueryUsageConfig(BaseUsageConfig):
@@ -316,7 +318,7 @@ def have_table_data_read_permission(self) -> bool:
     )
 
     max_threads_dataset_parallelism: int = Field(
-        default=SCHEMA_PARALLELISM,
+        default=DEFAULT_BQ_SCHEMA_PARALLELISM,
         description="Number of worker threads to use to parallelize BigQuery Dataset Metadata Extraction."
         " Set to 1 to disable.",
     )