Convert numpy MaskedArray to ensure mask is softened.
(data: ma.MaskedArray)
| 513 | |
| 514 | |
| 515 | def sanitize_masked_array(data: ma.MaskedArray) -> np.ndarray: |
| 516 | """ |
| 517 | Convert numpy MaskedArray to ensure mask is softened. |
| 518 | """ |
| 519 | mask = ma.getmaskarray(data) |
| 520 | if mask.any(): |
| 521 | dtype, fill_value = maybe_promote(data.dtype, np.nan) |
| 522 | dtype = cast(np.dtype, dtype) |
| 523 | data = ma.asarray(data.astype(dtype, copy=True)) |
| 524 | data.soften_mask() # set hardmask False if it was True |
| 525 | data[mask] = fill_value |
| 526 | else: |
| 527 | data = data.copy() |
| 528 | return data |
| 529 | |
| 530 | |
| 531 | def sanitize_array( |
no test coverage detected