FIX: modified pyart examples to handle new hydroclass_semisupervised …

…output
ARM-DOE · Aug 28, 2024 · 3ff109b · 3ff109b
1 parent a4f4fb0
commit 3ff109b
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 15 deletions.
diff --git a/examples/retrieve/plot_hydrometeor.py b/examples/retrieve/plot_hydrometeor.py
@@ -53,7 +53,7 @@
     kdp_field="specific_differential_phase",
     rhv_field="uncorrected_cross_correlation_ratio",
     temp_field="temperature",
-)
+)["hydro"]
 
 radar.add_field("radar_echo_classification", hydro)
 

diff --git a/examples/retrieve/plot_hydrometeor_class_x_band.py b/examples/retrieve/plot_hydrometeor_class_x_band.py
@@ -98,7 +98,7 @@
     rhv_field="RHOHV",
     temp_field="sounding_temperature",
     radar_freq=9.2e9,
-)
+)["hydro"]
 
 radar.add_field("hydro_classification", hydromet_class, replace_existing=True)
 

diff --git a/pyart/correct/phase_proc.py b/pyart/correct/phase_proc.py
@@ -108,7 +108,11 @@ def fzl_index(fzl, ranges, elevation, radar_height):
     p_r = 4.0 * Re / 3.0
     z = (
         radar_height
-        + (ranges**2 + p_r**2 + 2.0 * ranges * p_r * np.sin(elevation * np.pi / 180.0))
+        + (
+            ranges**2
+            + p_r**2
+            + 2.0 * ranges * p_r * np.sin(elevation * np.pi / 180.0)
+        )
         ** 0.5
         - p_r
     )

diff --git a/pyart/retrieve/qvp.py b/pyart/retrieve/qvp.py
@@ -24,7 +24,7 @@
     _update_qvp_metadata
     _update_along_coord_metadata
 
-"""
+""" "" "" "" "" "" "" "" "" "" "" "" "" "" ""
 
 from copy import deepcopy
 from warnings import warn
@@ -106,7 +106,6 @@ def quasi_vertical_profile(radar, desired_angle=None, fields=None, gatefilter=No
         fields = radar.fields
 
         for field in fields:
-
             # Filtering data based on defined gatefilter
             # If none is defined goes to else statement
             if gatefilter is not None:
@@ -431,7 +430,6 @@ def compute_rqvp(
         )
 
     for field_name in field_names:
-
         # mask weights where there is no data
         mask = np.ma.getmaskarray(val_interp[field_name])
         weight_aux = np.ma.masked_where(mask, weight)

diff --git a/pyart/util/radar_utils.py b/pyart/util/radar_utils.py
@@ -414,17 +414,17 @@ def subset_radar(
 
     if radar_aux.instrument_parameters is not None:
         if "nyquist_velocity" in radar_aux.instrument_parameters:
-            radar_aux.instrument_parameters["nyquist_velocity"]["data"] = (
-                radar_aux.instrument_parameters["nyquist_velocity"]["data"][ind_rays]
-            )
+            radar_aux.instrument_parameters["nyquist_velocity"][
+                "data"
+            ] = radar_aux.instrument_parameters["nyquist_velocity"]["data"][ind_rays]
         if "pulse_width" in radar_aux.instrument_parameters:
-            radar_aux.instrument_parameters["pulse_width"]["data"] = (
-                radar_aux.instrument_parameters["pulse_width"]["data"][ind_rays]
-            )
+            radar_aux.instrument_parameters["pulse_width"][
+                "data"
+            ] = radar_aux.instrument_parameters["pulse_width"]["data"][ind_rays]
         if "number_of_pulses" in radar_aux.instrument_parameters:
-            radar_aux.instrument_parameters["number_of_pulses"]["data"] = (
-                radar_aux.instrument_parameters["number_of_pulses"]["data"][ind_rays]
-            )
+            radar_aux.instrument_parameters["number_of_pulses"][
+                "data"
+            ] = radar_aux.instrument_parameters["number_of_pulses"]["data"][ind_rays]
 
     # Get new fields
     if field_names is None: