|
1 | 1 | """Create dataset.""" |
2 | 2 |
|
| 3 | +import numpy as np |
3 | 4 | import xarray as xr |
4 | 5 |
|
5 | 6 | from imap_processing.cdf.imap_cdf_manager import ImapCdfAttributes |
@@ -186,15 +187,29 @@ def create_dataset( # noqa: PLR0912 |
186 | 187 | attrs=cdf_manager.get_variable_attributes(key, check_schema=False), |
187 | 188 | ) |
188 | 189 | elif key in {"energy_range_edges"}: |
| 190 | + dataset["energy_range_edges_dim"] = xr.DataArray( |
| 191 | + np.arange(len(data)), |
| 192 | + dims=["energy_range_edges_dim"], |
| 193 | + attrs=cdf_manager.get_variable_attributes( |
| 194 | + "energy_range_edges_dim", check_schema=False |
| 195 | + ), |
| 196 | + ) |
189 | 197 | dataset[key] = xr.DataArray( |
190 | 198 | data, |
191 | | - dims=["energy_range_edges"], |
| 199 | + dims=["energy_range_edges_dim"], |
192 | 200 | attrs=cdf_manager.get_variable_attributes(key, check_schema=False), |
193 | 201 | ) |
194 | 202 | elif key in {"energy_range_flags"}: |
| 203 | + dataset["energy_range_flags_dim"] = xr.DataArray( |
| 204 | + np.arange(len(data)), |
| 205 | + dims=["energy_range_flags_dim"], |
| 206 | + attrs=cdf_manager.get_variable_attributes( |
| 207 | + "energy_range_flags_dim", check_schema=False |
| 208 | + ), |
| 209 | + ) |
195 | 210 | dataset[key] = xr.DataArray( |
196 | 211 | data, |
197 | | - dims=["energy_range_flags"], |
| 212 | + dims=["energy_range_flags_dim"], |
198 | 213 | attrs=cdf_manager.get_variable_attributes(key, check_schema=False), |
199 | 214 | ) |
200 | 215 | else: |
|
0 commit comments