fix: exporting nonexistent variables

CyanideCN · Jun 16, 2024 · b1a47b5 · b1a47b5
1 parent 1746f92
commit b1a47b5
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/cinrad/io/export.py b/cinrad/io/export.py
@@ -90,15 +90,21 @@ def standard_data_to_pyart(f: StandardData, radius: int = 460) -> pyart.core.Rad
     fields = {}
     nscans = f.get_nscans()
 
+    all_var = list()
+    for lvl in f.data:
+        all_var.extend(f.data[lvl].keys())
+    all_var_set = set(all_var)
+
     for mom in mapping.keys():
-        name = mapping[mom]
-        dic = filemetadata(name)
-        dic["_FillValue"] = pyart.config.get_fillvalue()
-        raw_arr = [f.get_raw(nel, radius, mom) for nel in range(nscans)]
-        sel_arr = [i if not isinstance(i, tuple) else i[0] for i in raw_arr]
-        moment_data = np.ma.vstack(sel_arr)
-        dic["data"] = moment_data
-        fields[name] = dic
+        if mom in all_var_set:
+            name = mapping[mom]
+            dic = filemetadata(name)
+            dic["_FillValue"] = pyart.config.get_fillvalue()
+            raw_arr = [f.get_raw(nel, radius, mom) for nel in range(nscans)]
+            sel_arr = [i if not isinstance(i, tuple) else i[0] for i in raw_arr]
+            moment_data = np.ma.vstack(sel_arr)
+            dic["data"] = moment_data
+            fields[name] = dic
 
     nyquist_velocity = filemetadata("nyquist_velocity")
     nyquist_velocity["data"] = np.array(