add check for numpy array

Thomasjkeel · Jan 16, 2025 · 8672f4c · 8672f4c
1 parent 16f7fb2
commit 8672f4c
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/jsmetrics/utils/spatial_utils.py b/jsmetrics/utils/spatial_utils.py
@@ -289,11 +289,9 @@ def get_one_contour_linestring(dataarray, contour_level):
     ), "Data array needs to have latitude and longitude coords"
     if "time" in dataarray.dims:
         dataarray = dataarray.squeeze("time")
-    try:
-        assert contour_level == False
-    except:
-        raise AssertionError(f"FAILED: {type(contour_level)}")
-    one_contour = dataarray.plot.contour(levels=[float(contour_level[0])])
+    if not isinstance(contour_level, np.ndarray):
+        contour_level = np.array(contour_level)
+    one_contour = dataarray.plot.contour(levels=[float(contour_level)])
     matplotlib.pyplot.close()
     one_contour_segments = seperate_one_contour_into_line_segments(
         one_contour.get_paths()[0]