Properly convert infinity and NaN from C++ to Python

NoahAmsel · Dec 4, 2020 · f11d925 · f11d925
1 parent f914da6
commit f11d925
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/include/pybind11_json/pybind11_json.hpp b/include/pybind11_json/pybind11_json.hpp
@@ -11,6 +11,7 @@
 
 #include <string>
 #include <vector>
+#include <cmath>
 
 #include "nlohmann/json.hpp"
 
@@ -34,7 +35,7 @@ namespace pyjson
         else if (j.is_number())
         {
             double number = j.get<double>();
-            if (number == std::floor(number))
+            if (isfinite(number) && number == std::floor(number))
             {
                 return py::int_(j.get<long>());
             }