def load_config()

in genai-on-vertex-ai/gemini/evals_playbook/utils/config.py [0:0]

• 26 lines of code
• 4 McCabe index (conditional complexity)

def load_config():

    config = configparser.ConfigParser()

    if is_root_dir():
        current_dir = os.getcwd()
        config.read(current_dir + '/config.ini')
        root_dir = current_dir
    else:
        root_dir = os.path.abspath(os.path.join(os.getcwd(), '..'))
        config.read(root_dir+'/config.ini')

    if not 'root_dir' in locals():  # If not found in any parent dir
        raise FileNotFoundError("config.ini not found in current or parent directories.")
        
    # Make variables global for modification
    global PROJECT_ID,LOCATION,STAGING_BUCKET,STAGING_BUCKET_URI,BQ_DATASET_ID,BQ_LOCATION,BQ_TABLES_SQL_PATH,BQ_PREFIX,BQ_T_EVAL_TASKS,BQ_T_EXPERIMENTS,BQ_T_PROMPTS,BQ_T_DATASETS,BQ_T_EVAL_RUN_DETAILS,BQ_T_EVAL_RUNS

    
    PROJECT_ID = config['GCP']['PROJECT_ID']
    LOCATION = config['GCP']['LOCATION']
    STAGING_BUCKET = config['CLOUDSTORAGE']['STAGING_BUCKET']
    STAGING_BUCKET_URI = config['CLOUDSTORAGE']['STAGING_BUCKET_URI']
    BQ_DATASET_ID = config['BIGQUERY']['BQ_DATASET_ID']
    BQ_LOCATION = config['BIGQUERY']['BQ_LOCATION']
    BQ_TABLES_SQL_PATH = config['BIGQUERY']['BQ_TABLES_SQL_PATH']
    BQ_PREFIX = config['BIGQUERY']['BQ_PREFIX']
    BQ_T_EVAL_TASKS = config['BIGQUERY']['BQ_T_EVAL_TASKS']
    BQ_T_EXPERIMENTS = config['BIGQUERY']['BQ_T_EXPERIMENTS']
    BQ_T_PROMPTS = config['BIGQUERY']['BQ_T_PROMPTS']
    BQ_T_DATASETS = config['BIGQUERY']['BQ_T_DATASETS']
    BQ_T_EVAL_RUN_DETAILS = config['BIGQUERY']['BQ_T_EVAL_RUN_DETAILS']
    BQ_T_EVAL_RUNS = config['BIGQUERY']['BQ_T_EVAL_RUNS']