Fix a unit test

Signed-off-by: Ruofei Zou <[email protected]>
amundsen-io · Oct 29, 2024 · 88e9f79 · 88e9f79
1 parent 4db4706
commit 88e9f79
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/common/amundsen_common/models/dashboard.py b/common/amundsen_common/models/dashboard.py
@@ -10,21 +10,22 @@
 
 class SafeFloat(fields.Field):
     def _serialize(self, value, attr, obj, **kwargs):
-        if value == '':
+        if value is None or value == '':
             return None
         try:
             return float(value)
         except ValueError:
             self.fail('invalid', input=value)
 
     def _deserialize(self, value, attr, data, **kwargs):
-        if value == '':
+        if value is None or value == '':
             return None
         try:
             return float(value)
         except ValueError:
             self.fail('invalid', input=value)
 
+
 @attr.s(auto_attribs=True, kw_only=True)
 class DashboardSummary:
     uri: str = attr.ib()