Index

A

add_units() (in module src.plot_utils) aggregate_matches() (in module src.wandb_utils) aggregate_table() (in module src.wandb_utils) animate_all() (src.models.ocean.Ocean method) animate_coupling() (in module src.visualisation.ani) animate_ds() (in module src.visualisation.ani) animate_qflx_diff() (in module src.visualisation.ani)

animate_xr_da() (in module src.visualisation.ani) archive_dir_from_config() (in module src.wandb_utils) arrow_plot() (in module src.visualisation.arrow) Atmos (class in src.models.atmos) atmos_bias() (in module src.visualisation.bias) atmos_input_file_path() (in module src.constants) axis_formatter() (in module src.plot_utils)

B

between_two() (in module src.search)

C

calculate_byte_size_recursively() (in module src.utils) calculate_cmip6_ts_trends() (in module src.visualisation.bias) calculate_nino3_4_from_noaa() (in module src.metrics) can_coords() (in module src.xr_utils) cd_heatmaps() (in module src.visualisation.sensitivity) cd_plots() (in module src.visualisation.sensitivity) cd_variation_comp() (in module src.wandb_utils) change_table() (in module src.wandb_utils) clear() (in module src.model_utils.clear) clim60_name() (src.models.model_setup.ModelSetup method) clim_file() (src.models.model_setup.ModelSetup method) clim_name() (src.models.model_setup.ModelSetup method) clip() (in module src.xr_utils) cmap() (in module src.plot_utils) cmip6_ensemble_var() (in module src.constants)

cmip6_file() (in module src.constants) comp_atm_prwnd() (in module src.visualisation.comp_v_seager19) comp_oc_htherm() (in module src.visualisation.comp_v_seager19) comp_oc_sst() (in module src.visualisation.comp_v_seager19) comp_plot() (in module src.visualisation.comp_v_seager19) comp_prcp_quiver_plot() (in module src.visualisation.comp_v_seager19) comp_seq() (in module src.visualisation.comp_seq) comp_uc_atm() (in module src.visualisation.comp_v_seager19) comp_uc_oc() (in module src.visualisation.comp_v_seager19) compile_all() (src.models.ocean.Ocean method) copy_old_io() (src.models.ocean.Ocean method) Coupling (class in src.models.coupling) coupling_frame() (in module src.visualisation.convergence) coupling_video() (src.models.model_setup.ModelSetup method) cut_and_taper() (in module src.xr_utils)

D

da_diagnostics() (in module src.test_utils.investigate_inputs) derived_param() (in module src.configs.config) didnt_blow_up() (in module src.wandb_utils) dq_df() (src.models.model_setup.ModelSetup method)

dq_df_file() (src.models.model_setup.ModelSetup method) dq_dt() (src.models.model_setup.ModelSetup method) dq_dtemp_file() (src.models.model_setup.ModelSetup method) dq_output() (src.models.model_setup.ModelSetup method)

E

ecmwf_sfcwind() (src.models.model_setup.ModelSetup method) edit_inputs() (src.models.ocean.Ocean method)

edit_run() (src.models.ocean.Ocean method) eps_heatmaps() (in module src.visualisation.sensitivity) eps_plots() (in module src.visualisation.sensitivity)

F

f_cor() (src.models.atmos.Atmos method) f_dqlh_dtemp() (src.models.atmos.Atmos method) f_dqlw_df() (src.models.atmos.Atmos method) f_dqlw_dtemp() (src.models.atmos.Atmos method) f_dqs_dtemp() (src.models.atmos.Atmos method) f_ebar() (src.models.atmos.Atmos method) f_es() (src.models.atmos.Atmos method) f_evap() (src.models.atmos.Atmos method) f_mc() (src.models.atmos.Atmos method) f_qa() (src.models.atmos.Atmos method) f_qa2() (src.models.atmos.Atmos method) f_qlh() (src.models.atmos.Atmos method) f_qlw() (src.models.atmos.Atmos method)

f_qlw1() (src.models.atmos.Atmos method) f_qlw2() (src.models.atmos.Atmos method) f_qs() (src.models.atmos.Atmos method) f_stress() (src.models.coupling.Coupling method) f_temp_a() (src.models.atmos.Atmos method) field_plot() (in module src.visualisation.comp_v_seager19) FileNames (class in src.data_loading.ecmwf) final_coup_plot() (in module src.visualisation.convergence) find_missing() (in module src.wandb_utils) finished_names() (in module src.wandb_utils) fit() (in module src.models.poly) fix_calendar() (in module src.xr_utils) fix_config() (in module src.wandb_utils) format_config() (in module src.configs.config)

G

get_and_unzip() (in module src.data_loading.download) get_byte_size() (in module src.utils) get_clim() (in module src.xr_utils) get_cloud_const() (src.models.atmos.Atmos method) get_cmip6_mmm() (in module src.data_loading.download) get_cmip6_mmm_clim60() (in module src.data_loading.download) get_data() (in module src.data_loading.download) get_dclim() (src.models.atmos.Atmos method) get_default_setup() (in module src.utils) get_dim() (in module src.plot_utils) get_era5() (in module src.data_loading.ecmwf) get_ersstv5() (in module src.data_loading.psl) get_figure_data() (in module src.data_loading.download) get_full_csv() (in module src.wandb_utils) get_main_variables() (in module src.data_loading.ecmwf) get_mekong_variables() (in module src.data_loading.ecmwf)

get_member_data() (in module src.data_loading.download) get_mmm() (in module src.data_loading.download) get_nino_trend() (in module src.visualisation.nino) get_noaa_data() (in module src.data_loading.download) get_original_models() (in module src.data_loading.download) get_other_trends() (in module src.metrics) get_psl_indices() (in module src.data_loading.psl) get_tau_anom() (src.models.coupling.Coupling method) get_test_nino_data() (in module src.data_loading.download) get_trend() (in module src.xr_utils) get_trends() (in module src.visualisation.bias) get_ts_ensemble() (in module src.data_loading.download) get_uv() (in module src.data_loading.download) get_v() (in module src.wandb_utils) get_wandb_data() (in module src.wandb_utils) get_zip() (in module src.data_loading.zenodo)

H

hr_time() (in module src.utils)

human_readable_size() (in module src.utils)

I

in_notebook() (in module src.utils)

K

k_plots() (in module src.visualisation.sensitivity)

L

label_subplots() (in module src.plot_utils) line_plot() (in module src.visualisation.bias) linear_qflx_replacement() (in module src.data_loading.ingrid) list_to_hydra_input() (in module src.search)

load_clim60() (src.models.atmos.Atmos method) load_config() (in module src.configs.load_config) load_noaa_data() (in module src.metrics) log() (src.models.coupling.Coupling method)

M

main() (in module src.main) (in module src.search) make_figure() (src.models.atmos.Atmos method) make_plots() (in module src.visualisation.nino) make_plots_example() (in module src.visualisation.comp_v_seager19) make_results_xr() (in module src.visualisation.arrow) mask_file() (src.models.model_setup.ModelSetup method) mem_to_dict() (in module src.model_utils.mem_to_input) mems_to_df() (in module src.model_utils.mem_to_input) metric_conv_data() (in module src.wandb_utils) metric_conv_plot() (in module src.visualisation.convergence) min_clim() (in module src.xr_utils) ModelSetup (class in src.models.model_setup) module src src.configs src.configs.config src.configs.load_config src.configs.pangeo src.constants src.data_loading src.data_loading.download src.data_loading.ecmwf src.data_loading.ingrid src.data_loading.psl src.data_loading.zenodo src.main src.metrics src.model_utils src.model_utils.clear src.model_utils.mem_to_input src.models src.models.atmos src.models.coupling src.models.model_setup src.models.ocean src.models.poly src.plot_utils src.search src.test_utils src.test_utils.investigate_inputs src.utils src.visualisation src.visualisation.ani src.visualisation.arrow src.visualisation.bias src.visualisation.comp_cd src.visualisation.comp_seq src.visualisation.comp_v_seager19 src.visualisation.convergence src.visualisation.nino src.visualisation.pairplot src.visualisation.quiver src.visualisation.sensitivity src.visualisation.trends src.wandb_utils src.xr_utils

multi_bias_plot() (in module src.visualisation.bias) multi_panel_nino() (in module src.visualisation.nino)

N

nino_calculate() (in module src.metrics) nino_nc() (src.models.model_setup.ModelSetup method)

nino_png() (src.models.model_setup.ModelSetup method) nummode_plots() (in module src.visualisation.sensitivity)

O

Ocean (class in src.models.ocean) ocean_input_file_path() (in module src.constants) om_mask() (src.models.model_setup.ModelSetup method) om_run2f_nc() (src.models.model_setup.ModelSetup method) open_dataarray() (in module src.xr_utils)

open_dataset() (in module src.xr_utils) origninal_arrow_plot() (in module src.visualisation.arrow) output_dq() (src.models.atmos.Atmos method) output_fig_2_data() (in module src.wandb_utils) output_trends() (src.models.atmos.Atmos method)

P

pairplot() (in module src.visualisation.pairplot) plot() (in module src.models.poly) plot_arrow_plot_6() (in module src.visualisation.arrow) plot_bias() (in module src.visualisation.bias) plot_comp_cd() (in module src.visualisation.comp_cd) plot_nino() (in module src.visualisation.nino) plot_nino_box() (in module src.visualisation.nino)

plot_results_xr() (in module src.visualisation.arrow) pqp_part() (in module src.visualisation.quiver) prcp_quiver_plot() (in module src.visualisation.quiver) (src.models.model_setup.ModelSetup method) ps_defaults() (in module src.plot_utils) psl_index_da() (in module src.data_loading.psl) psl_metric_test() (in module src.data_loading.psl)

Q

q_output() (src.models.model_setup.ModelSetup method)

R

rem_var() (in module src.xr_utils) remainder_combinations() (in module src.search) rename_mekong() (in module src.data_loading.ecmwf) rep_plot() (src.models.model_setup.ModelSetup method) replace_dq() (src.models.coupling.Coupling method) replace_item() (in module src.models.ocean) replace_nino3_4_from_noaa() (in module src.metrics) replace_stress() (src.models.coupling.Coupling method)

replace_surface_temp() (src.models.coupling.Coupling method) return_figure_ds() (in module src.visualisation.comp_v_seager19) return_seager19_ds() (in module src.visualisation.comp_v_seager19) return_var_list() (in module src.visualisation.comp_v_seager19) run() (src.models.coupling.Coupling method) (src.models.ocean.Ocean method) run_all() (src.models.atmos.Atmos method) (src.models.ocean.Ocean method) run_path() (in module src.constants)

S

s91_solver() (src.models.atmos.Atmos method) sel() (in module src.xr_utils) set_dim() (in module src.plot_utils) setup_from_config() (in module src.wandb_utils) setup_from_name() (in module src.wandb_utils) smooth121() (src.models.atmos.Atmos method) spatial_mean() (in module src.xr_utils) src module src.configs module src.configs.config module src.configs.load_config module src.configs.pangeo module src.constants module src.data_loading module src.data_loading.download module src.data_loading.ecmwf module src.data_loading.ingrid module src.data_loading.psl module src.data_loading.zenodo module src.main module src.metrics module src.model_utils module src.model_utils.clear module src.model_utils.mem_to_input module src.models module src.models.atmos module src.models.coupling module src.models.model_setup module src.models.ocean module

src.models.poly module src.plot_utils module src.search module src.test_utils module src.test_utils.investigate_inputs module src.utils module src.visualisation module src.visualisation.ani module src.visualisation.arrow module src.visualisation.bias module src.visualisation.comp_cd module src.visualisation.comp_seq module src.visualisation.comp_v_seager19 module src.visualisation.convergence module src.visualisation.nino module src.visualisation.pairplot module src.visualisation.quiver module src.visualisation.sensitivity module src.visualisation.trends module src.wandb_utils module src.xr_utils module sst_file() (src.models.model_setup.ModelSetup method) sst_replacement_file() (src.models.model_setup.ModelSetup method) start_wandb() (in module src.wandb_utils) stress_clim_file() (src.models.model_setup.ModelSetup method) stress_file() (src.models.model_setup.ModelSetup method) sub_main() (in module src.main) summary_table() (in module src.wandb_utils)

T

tau_anom_ds() (src.models.coupling.Coupling method) tau_base() (src.models.model_setup.ModelSetup method) tau_clim_base() (src.models.model_setup.ModelSetup method) tau_clim_x() (src.models.model_setup.ModelSetup method) tau_clim_y() (src.models.model_setup.ModelSetup method) tau_x() (src.models.model_setup.ModelSetup method) tau_y() (src.models.model_setup.ModelSetup method) tcam_output() (src.models.model_setup.ModelSetup method) tdma_solver() (src.models.atmos.Atmos method) temperature_line_plot() (in module src.visualisation.bias) terminal_call() (in module src.search)

tex_uf() (in module src.plot_utils) time_limit() (in module src.utils) time_stamp() (in module src.utils) time_title() (in module src.plot_utils) timeit() (in module src.utils) TimeoutException trends_from_csv() (in module src.visualisation.bias) trends_from_da() (in module src.visualisation.bias) ts_clim() (src.models.model_setup.ModelSetup method) ts_clim60() (src.models.model_setup.ModelSetup method) ts_trend() (src.models.model_setup.ModelSetup method) tuq_trend_plot() (src.models.model_setup.ModelSetup method)

U

up_therm_qnet() (in module src.visualisation.trends)

V

var_clt_combinations() (in module src.search)

var_ts_combinations() (in module src.search) variable_combinations() (in module src.search)

W

which_comp() (in module src.search)