fix geodiff to always return floats in dict

AndrewAnnex · Feb 8, 2023 · e81da4b · e81da4b
1 parent 9757e0b
commit e81da4b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/asap_stereo/asap.py b/src/asap_stereo/asap.py
@@ -913,6 +913,14 @@ def get_geo_diff(self, ref_dem, src_dem=None):
                     'StdDev of difference': stats['stdDev'],
                     'Median difference': stats["mean"], # yes I know this isn't correct but gdal doens't compute this for us
                 }
+            for k, v in res.items():
+                try:
+                    res[k] = float(v)
+                except ValueError:
+                    try:
+                        res[k] = float(''.join(re.findall(r'-?\d+\.?\d+', v))) # this won't grab all floats like Nans or si notation
+                    except ValueError:
+                        res[k] = 0.0 
             return res
 
     def estimate_max_disparity(self, ref_dem, src_dem=None):