fix float nan comparison

This change also adds minor docstrings fixes and bumps pre-commit blac
version.
This commit is contained in:
James Hilton-Balfe
2022-04-24 00:13:58 +01:00
committed by GitHub
parent ac96d8254b
commit 6f7d706a8e

View File

@@ -434,7 +434,7 @@ def _parse_float(value: Any) -> float:
Parameters
----------
value : Any
value: Any
Value to parse
Returns
@@ -456,20 +456,19 @@ def _dump_float(value: float) -> Union[float, str]:
Parameters
----------
value : float
value: float
Value to dump
Returns
-------
Union[float, str]
Dumped valid, either a float or the strings
"Infinity" or "-Infinity"
Dumped value, either a float or the strings
"""
if value == float("inf"):
return INFINITY
if value == -float("inf"):
return NEG_INFINITY
if value == float("nan"):
if isinstance(value, float) and math.isnan(value):
return NAN
return value