Update plot_qpe.py

examples/retrieve/plot_qpe.py

@@ -29,11 +29,11 @@
 import pytz
 import xarray as xr
 
-import pyart
+import 
 
 
 def column_profile(
-    radar: pyart.core.Radar,
+    radar: .core.Radar,
     latitude: float = 33.8,
     longitude: float = -88.3,
     azimuth_spread: float = 3,
@@ -52,7 +52,7 @@ def column_profile(
 
     Parameters
     ----------
-    radar : pyart.core.Radar
+    radar : .core.Radar
         Py-ART radar object containing volume scan data.
     latitude : float, optional
         Latitude of the point of interest (default is 33.8).
@@ -77,7 +77,7 @@ def column_profile(
     if min_alt is None or max_alt is None:
         raise ValueError("Both min_alt and max_alt must be specified.")
 
-    col_prof = pyart.util.column_vertical_profile(
+    col_prof = .util.column_vertical_profile(
         radar,
         latitude=latitude,
         longitude=longitude,
@@ -142,7 +142,7 @@ def download_nexrad(timezone, date, site, local_date=False):
 rain_rate_list = []
 
 for file in files:
-    radar = pyart.io.read_nexrad_archive("s3://" + file)
+    radar = .io.read_nexrad_archive("s3://" + file)
     col_prof_interp = column_profile(
         radar,
         latitude=33.5,
@@ -160,7 +160,7 @@ def download_nexrad(timezone, date, site, local_date=False):
 da_rain_rate = xr.concat(rain_rate_list, dim="time")
 
 da_rain_rate.plot.contourf(
-    x="time", levels=range(0, 26, 2), cmap="pyart_RRate11", figsize=[10, 3]
+    x="time", levels=range(0, 26, 2), cmap="RRate11", figsize=[10, 3]
 )
 plt.title("QPE")
 plt.xlabel("Time")