code cleanup

GEOS-ESM · Jan 6, 2025 · 774bbd9 · 774bbd9
1 parent 181dcb7
commit 774bbd9
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/Tests/generate_extdatadriver_input.py b/Tests/generate_extdatadriver_input.py
@@ -5,6 +5,8 @@
 import yaml
 import glob
 
+dims_dict = {"2":"xy", "3":"xyz"}
+
 def get_dims(component_map, name):
     for comp in component_map:
         if name in component_map[comp]:
@@ -114,10 +116,7 @@ def parse_args():
          long_name = "NA"
          units = "NA"
          dims = get_dims(component_map, item) 
-         if dims == "2":
-            cdims = "xy"
-         elif dims == "3":
-            cdims = "xyz"
+         cdims = dims_dict[dims]
 
          if item not in written:
             f_agcm.write(item+cm+long_name+cm+units+cm+cdims+cm+"c"+nl)
@@ -134,10 +133,7 @@ def parse_args():
              long_name = "NA"
              units = "NA"
              dims = get_dims(component_map, item) 
-             if dims == "2":
-                cdims = "xy"
-             elif dims == "3":
-                cdims = "xyz"
+             cdims = dims_dict[dims]
 
              if item not in written:
                 f_agcm.write(item+cm+long_name+cm+units+cm+cdims+cm+"c"+nl)