Add ocean variables file, light code formatting

evv4esm/extensions/kso.py

@@ -52,17 +52,16 @@
 import livvkit
 import numpy as np
 import pandas as pd
+from evv4esm import EVVException, human_color_names
+from evv4esm.ensembles import e3sm
+from evv4esm.ensembles.tools import prob_plot
+from evv4esm.utils import bib2html
 from livvkit import elements as el
 from livvkit.util import functions as fn
 from livvkit.util.LIVVDict import LIVVDict
 from scipy import stats
 from statsmodels.stats import multitest as smm
 
-from evv4esm import EVVException, human_color_names
-from evv4esm.ensembles import e3sm
-from evv4esm.ensembles.tools import prob_plot
-from evv4esm.utils import bib2html
-
 
 def variable_set(name):
     var_sets = fn.read_json(os.path.join(os.path.dirname(__file__), "ocean_vars.json"))
@@ -154,7 +153,7 @@ def parse_args(args=None):
     if args.config:
         default_args = parser.parse_args([])
 
-        for key, val, in vars(args).items():
+        for key, val in vars(args).items():
             if val != vars(default_args)[key]:
                 args.config["ks"][key] = val
 

evv4esm/extensions/ocean_vars.json

@@ -0,0 +1,9 @@
+{
+    "default": [
+        "activeTracers{}_salinity",
+        "activeTracers{}_temperature",
+        "ssh{}",
+        "velocityMeridional{}",
+        "velocityZonal{}"
+    ]
+}