From 8672f4cbe4a42ae78c97378d61f916993fbde0d0 Mon Sep 17 00:00:00 2001 From: Thomasjkeel Date: Thu, 16 Jan 2025 16:17:44 +0000 Subject: [PATCH] add check for numpy array --- jsmetrics/utils/spatial_utils.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/jsmetrics/utils/spatial_utils.py b/jsmetrics/utils/spatial_utils.py index 02edc54..1258cbd 100644 --- 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]