server: add convert inf to null in variables

factly · Jul 5, 2024 · 502b8a4 · 502b8a4
1 parent 2e78428
commit 502b8a4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/app/models/variables.py b/app/models/variables.py
@@ -1,4 +1,4 @@
-from math import isnan
+from math import isinf, isnan
 from typing import Any, Dict, List, Optional, Union
 
 from pydantic import BaseModel, Field, constr, validator
@@ -57,6 +57,8 @@ class VariableProperties(BaseModel):
     def change_nan_to_none(cls, v, field):
         if field.outer_type_ is float and isnan(v):
             return None
+        elif isinstance(v, float) and isinf(v):
+            return None
         return v