Code update

PiperOrigin-RevId: 717385750
google-research · Jan 20, 2025 · 5c1c58a · 5c1c58a
1 parent fed21f3
commit 5c1c58a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ics/projects/probabilistic_diffusion/downscaling/gcm_wrf/input_pipelines/paired_hourly.py b/...ics/projects/probabilistic_diffusion/downscaling/gcm_wrf/input_pipelines/paired_hourly.py
@@ -149,7 +149,7 @@ def __init__(
 
     self._dates = data_utils.get_common_times(input_ds, date_range)
     self._len = input_ds.dims["time"]
-    self._time_array = xr_.read(input_ds["time"]).data
+    self._time_array = xrts.read(input_ds["time"]).data
     self._resample_at_nan = resample_at_nan
     self._resample_seed = resample_seed
     self.use_temporal_inputs = use_temporal_inputs