import os
from acme_diags.parameter.core_parameter import CoreParameter
from acme_diags.parameter.arm_diags_parameter import ARMDiagsParameter
from acme_diags.run import runner

param = CoreParameter()

param.test_data_path = '/global/cscratch1/sd/chengzhu/arm_diags_e3sm/test'
param.test_name = 'NGD_convection'
param.reference_data_path = '/global/cfs/cdirs/e3sm/e3sm_diags/obs_for_e3sm_diags/arm-diags-data'
param.ref_name = 'armdiags'
param.run_type = 'model_vs_obs'
prefix = '/global/cfs/cdirs/e3sm/www/zhang40/test_arm_ChrisT/'
#param.multiprocessing = True
#param.num_workers = 4
param.results_dir = os.path.join(prefix, 'arm_diags_v2_m2o')
#param.test_timeseries_input = True
#param.ref_timeseries_input = True
param.test_start_yr = '1991'
param.test_end_yr = '2010'
param.ref_start_yr = '0001'
param.ref_end_yr = '0001'

runner.sets_to_run = ['arm_diags']
runner.run_diags([param])