Skip to content

Commit 2075924

Browse files
Include metrics for model only run as in v2 (#949)
1 parent f831bc0 commit 2075924

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

e3sm_diags/driver/lat_lon_driver.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -579,6 +579,7 @@ def _create_metrics_dict(
579579
"min": var_test.min().item(),
580580
"max": var_test.max().item(),
581581
"mean": spatial_avg(ds_test, var_key), # type: ignore
582+
"std": std(ds_test, var_key),
582583
},
583584
"unit": ds_test[var_key].attrs["units"],
584585
}
@@ -591,6 +592,7 @@ def _create_metrics_dict(
591592
"min": var_ref.min().item(),
592593
"max": var_ref.max().item(),
593594
"mean": spatial_avg(ds_ref, var_key), # type: ignore
595+
"std": std(ds_ref, var_key),
594596
}
595597

596598
if ds_test_regrid is not None and ds_ref_regrid is not None:
@@ -615,6 +617,10 @@ def _create_metrics_dict(
615617
"corr": correlation(ds_test_regrid, ds_ref_regrid, var_key),
616618
}
617619

620+
# for model-only run
621+
if ds_test is not None and ds_ref_regrid is None:
622+
metrics_dict["test_regrid"] = metrics_dict["test"]
623+
618624
if ds_diff is not None:
619625
var_diff = ds_diff[var_key]
620626

0 commit comments

Comments
 (0)