modified survey module to remove numpy DeprecationWarning

jonnymaserati · Nov 12, 2023 · c8efa1b · c8efa1b
1 parent d0a1afc
commit c8efa1b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/welleng/survey.py b/welleng/survey.py
@@ -1343,6 +1343,9 @@ def _interpolate_survey(survey, x=0, index=0):
             2), with the interpolated station between them (index 1)
 
     """
+    if isinstance(index, np.ndarray):
+        index = index[0]
+
     index = int(index)
 
     assert index < len(survey.md) - 1, "Index is out of range"