issue #207 change speed compute formula to avoid mask shrinking

AntSimi · Sep 27, 2023 · 73b017c · 73b017c
1 parent e659c50
commit 73b017c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/py_eddy_tracker/dataset/grid.py b/src/py_eddy_tracker/dataset/grid.py
@@ -38,6 +38,7 @@
     round_,
     sin,
     sinc,
+    sqrt,
     where,
     zeros,
 )
@@ -1919,7 +1920,8 @@ def speed_coef_mean(self, contour):
 
     def init_speed_coef(self, uname="u", vname="v"):
         """Draft"""
-        self._speed_ev = (self.grid(uname) ** 2 + self.grid(vname) ** 2) ** 0.5
+        u, v = self.grid(uname), self.grid(vname)
+        self._speed_ev = sqrt(u * u + v * v)
 
     def display(self, ax, name, factor=1, ref=None, **kwargs):
         """