From a4744e38e59c562fbf7af9f5585d9050d5616ba4 Mon Sep 17 00:00:00 2001 From: Sean Freeman Date: Fri, 30 Aug 2024 14:45:56 -0500 Subject: [PATCH] fix for odd ns/us datetime conversions. converting to datetime64 always forces a us conversion, this only converts if we are in iris mode. --- tobac/segmentation/watershed_segmentation.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tobac/segmentation/watershed_segmentation.py b/tobac/segmentation/watershed_segmentation.py index f49099f4..b2fa7a54 100644 --- a/tobac/segmentation/watershed_segmentation.py +++ b/tobac/segmentation/watershed_segmentation.py @@ -1255,7 +1255,12 @@ def segmentation( features_out_list = [] # Iris workaround: convert cftime to datetime64 - all_times = features["time"].map(np.datetime64) + + if np.issubdtype(features["time"].dtype, np.datetime64): + # we are (likely) a numpy datetime + all_times = features["time"] + else: + all_times = features["time"].map(np.datetime64) for i_time, time_i in enumerate(field.coords[time_var_name]): field_at_time = field.isel({time_var_name: i_time})