vertical_mixing: fix for depths below -255m

OpenDrift · Dec 13, 2021 · 7c49eda · 7c49eda
1 parent 7930fe8
commit 7c49eda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opendrift/models/oceandrift.py b/opendrift/models/oceandrift.py
@@ -446,7 +446,7 @@ def vertical_mixing(self, store_depths=False):
             w = self.elements.terminal_velocity
 
             # Diffusivity and its gradient at z
-            zi = np.round(z_index(-self.elements.z)).astype(np.uint8)
+            zi = np.round(z_index(-self.elements.z)).astype(np.uint16)
             Kz = Kprofiles[zi, range(Kprofiles.shape[1])]
             dKdz = gradK[zi, range(Kprofiles.shape[1])]