I think you're looking for the errstate context manager: https://numpy.org/doc/stable/reference/generated/numpy.errstate.html
On Thu, May 9, 2024 at 1:11 PM <[email protected]> wrote: > The current way (according to 1.26 doc) of setting and resetting error is > ``` > old_settings = np.seterr(all='ignore') #seterr to known value > np.seterr(over='raise') > {'divide': 'ignore', 'over': 'ignore', 'under': 'ignore', 'invalid': > 'ignore'} > np.seterr(**old_settings) # reset to default > ``` > This may be tedious and not elegant when we need to suppress the error for > some certain lines, for example, `np.nan_to_num(a/b) ` as we need to > suppress divide here. > > I think it would be way more elegant to use `with` statement here, which > should be able to be implemented with some simple changes. > > An ideal result would be like: > ``` > with np.seterr(divide='ignore'): > np.nan_to_num(a/b) # no warning > > a/0 # still warn > ``` > _______________________________________________ > NumPy-Discussion mailing list -- [email protected] > To unsubscribe send an email to [email protected] > https://mail.python.org/mailman3/lists/numpy-discussion.python.org/ > Member address: [email protected] >
_______________________________________________ NumPy-Discussion mailing list -- [email protected] To unsubscribe send an email to [email protected] https://mail.python.org/mailman3/lists/numpy-discussion.python.org/ Member address: [email protected]
