diff --git a/examples/retrieve/plot_qpe.py b/examples/retrieve/plot_qpe.py index 96e67057c8..63e5df94aa 100644 --- a/examples/retrieve/plot_qpe.py +++ b/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")