Refactored main.py to build spherical grids as well as homogeneous ones.

import os

import numpy as np

from itertools import product

from assets.commons import (load_config_file,

                            parse_grid_overrides)

def build_model_grid():

    stg_config = load_config_file(os.path.join('stg', 'config', 'config.yml'))

    pl_density_index = float(stg_config['density_powerlaw_idx'])

    _model_type = 'spherical' if pl_density_index != 0 else 'homogeneous'

    config = load_config_file(os.path.join('config', 'config.yml'))

    # grid definition

    dust_temperatures = parse_grid_overrides(par_name='dust_temperature',

                                             config=config)

    central_densities = parse_grid_overrides(par_name='gas_density',

    densities = parse_grid_overrides(par_name='gas_density',

                                             config=config)

    lines = config['overrides']['lines_to_process']

    grid_tarfiles = []

    for (tdust, nh2) in product(dust_temperatures, central_densities):

    density_keyword = 'central_density' if _model_type == 'homogeneous' else 'density_value_at_reference'

    for (tdust, nh2) in product(dust_temperatures, densities):

        overrides = {

            'grid': {

                'dust_temperature': tdust,

                'central_density': nh2,

                density_keyword: nh2,

            }

        }

        grid_tarfiles.append(stg_main(override_config=overrides))