Bug fixes for compare_diags.py example script

gcpy/examples/diagnostics/compare_diags.py - Import rename_and_flip_gchp_rst_vars function from gcpy.util gcpy/examples/diagnostics/compare_diags.yml - Added "n_cores: -1" to specify maximum parallelization CHANGELOG.md - Updated accordingly Signed-off-by: Bob Yantosca <[email protected]>
geoschem · Dec 21, 2023 · 0b9b342 · 0b9b342
1 parent a056352
commit 0b9b342
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,6 +4,11 @@ All notable changes to GCPy will be documented in this file.
 
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [Unreleased] - TBD
+### Fixed
+- Import error in `gcpy/examples/diagnostics/compare_diags.py`
+- Added missing `n_cores` to `gcpy/examples/diagnostics/compare_diags.yml`
+
 ## [1.4.1] - 2023-12-08
 ### Fixed
 - Now use the proper default value for the `--weightsdir` argument to `gcpy/file_regrid.py`

diff --git a/gcpy/examples/diagnostics/compare_diags.py b/gcpy/examples/diagnostics/compare_diags.py
@@ -11,7 +11,7 @@
 import warnings
 import numpy as np
 from gcpy.util import add_missing_variables, compare_varnames, \
-    dataset_reader, read_config_file
+    dataset_reader, read_config_file, rename_and_flip_gchp_rst_vars
 from gcpy.constants import skip_these_vars
 from gcpy.plot.compare_single_level import compare_single_level
 from gcpy.plot.compare_zonal_mean import compare_zonal_mean
@@ -109,8 +109,8 @@ def read_data(config):
     # If the data is from a GCHP restart file, rename variables and
     # flip levels to match the GEOS-Chem Classic naming and level
     # conventions.  Otherwise no changes will be made.
-    refdata = util.rename_and_flip_gchp_rst_vars(refdata)
-    devdata = util.rename_and_flip_gchp_rst_vars(devdata)
+    refdata = rename_and_flip_gchp_rst_vars(refdata)
+    devdata = rename_and_flip_gchp_rst_vars(devdata)
 
     # Define dictionary for return
     data = {

diff --git a/gcpy/examples/diagnostics/compare_diags.yml b/gcpy/examples/diagnostics/compare_diags.yml
@@ -33,3 +33,4 @@ options:
     filename: ''
     skip_small_diffs: True
     small_diff_threshold: 0.0000
+  n_cores: -1