ENH: robustness to NA