improve runtime behavior

contrailcirrus · Nov 22, 2024 · 814f822 · 814f822
1 parent 1ffdba6
commit 814f822
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycontrails/datalib/ecmwf/era5.py b/pycontrails/datalib/ecmwf/era5.py
@@ -541,7 +541,7 @@ def _preprocess_era5_dataset(self, ds: xr.Dataset) -> xr.Dataset:
 
         # For "reanalysis-era5-single-levels"
         # then the netcdf file does not contain the dimension "level"
-        if "single-levels" in self.dataset:
+        if self.is_single_level:
             ds = ds.expand_dims(level=self.pressure_levels)
 
         # New CDS-Beta gives "valid_time" instead of "time"