fix: clamp nones into normal values as well

encord-team · Oct 3, 2023 · 1f988da · 1f988da
1 parent b19cf89
commit 1f988da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/encord_active/db/migrations/versions/2023_09_05_162048-v1_0.py b/src/encord_active/db/migrations/versions/2023_09_05_162048-v1_0.py
@@ -1369,8 +1369,8 @@ def _transform_char8(value: Optional[str]) -> Optional[int]:
     return int.from_bytes(value_bytes, byteorder="little", signed=True)
 
 
-def _transform_clamp_normal(value: float) -> float:
-    if math.isnan(value):
+def _transform_clamp_normal(value: Optional[float]) -> float:
+    if not value or math.isnan(value):
         return 0.0
     elif math.isinf(value):
         return 1.0 if value > 0.0 else 0.0