import os
import numpy
from e3sm_diags.parameter.core_parameter import CoreParameter


from e3sm_diags.run import runner

short_name = 't.se62-mar27.F2010-SCREAMv1.ne256pg2_ne256pg2.cntl'
test_ts = ''
start_yr = int('0001')
end_yr = int('0001')
num_years = end_yr - start_yr + 1

param = CoreParameter()

# Model
param.test_data_path = 'climo'
param.test_name = 't.se62-mar27.F2010-SCREAMv1.ne256pg2_ne256pg2.pm-gpu.n096t4xX.L128.vth200'
param.short_test_name = short_name

# Obs
param.reference_data_path = '/global/cfs/cdirs/e3sm/e3sm_diags/obs_for_e3sm_diags/climatology'


# Output dir
param.results_dir = 'model_vs_obs_0001-0001'

# Additional settings
param.run_type = 'model_vs_obs'
param.diff_title = 'Model - Observations'
param.output_format = ['png']
param.output_format_subplot = []
param.multiprocessing = True
param.num_workers = 6
#param.fail_on_incomplete = True
params = [param]

# Run
runner.sets_to_run = ['lat_lon', 'zonal_mean_xy', 'zonal_mean_2d',]
runner.run_diags(params)