Skip to content

Commit f07738d

Browse files
committed
Fix DC API key config
1 parent 74311f7 commit f07738d

2 files changed

Lines changed: 3 additions & 0 deletions

File tree

import-automation/executor/app/configs.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ class ExecutorConfig:
5757
storage_dev_bucket_name: str = 'unresolved_mcf'
5858
# DataCommons API key
5959
dc_api_key: str = ''
60+
autopush_dc_api_key: str = ''
6061
# Gemini API key
6162
gemini_api_key: str = ''
6263
# Executor output prefix in the storage_dev_bucket_name bucket.

import-automation/executor/main.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,8 @@ def run_import_job(absolute_import_name: str, import_config: str):
111111
logging.info(f'Import config: {config}')
112112
if config.dc_api_key:
113113
os.environ['DC_API_KEY'] = config.dc_api_key
114+
if config.autopush_dc_api_key:
115+
os.environ['AUTOPUSH_DC_API_KEY'] = config.autopush_dc_api_key
114116
executor = import_executor.ImportExecutor(
115117
uploader=file_uploader.GCSFileUploader(
116118
project_id=config.gcs_project_id,

0 commit comments

Comments
 (0)