handling special case where no previous times available for new forecast

oceanmodeling · Jul 11, 2024 · 90679be · 90679be
1 parent d43edea
commit 90679be
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/stormevents/nhc/track.py b/stormevents/nhc/track.py
@@ -1083,9 +1083,14 @@ def __compute_velocity(data: DataFrame) -> DataFrame:
                 shift_time = advisory_data.loc[ind[1], "datetime"]
                 if shift_time > this_time:
                     # update shift index
-                    shifted_indices[counter] = advisory_data["datetime"][
-                        advisory_data["datetime"] < this_time
-                    ].index[-1]
+                    if (advisory_data["datetime"] < this_time).sum() == 0:
+                        shifted_indices[counter] = advisory_data["datetime"][
+                            advisory_data["datetime"] > this_time
+                        ].index[0]
+                    else:
+                        shifted_indices[counter] = advisory_data["datetime"][
+                            advisory_data["datetime"] < this_time
+                        ].index[-1]
 
             _, inverse_azimuths, distances = geodetic.inv(
                 advisory_data.loc[indices, "longitude"],
@@ -1096,8 +1101,10 @@ def __compute_velocity(data: DataFrame) -> DataFrame:
 
             intervals = (
                 (
-                    advisory_data.loc[indices, "datetime"].values
-                    - advisory_data.loc[shifted_indices, "datetime"].values
+                    abs(
+                        advisory_data.loc[indices, "datetime"].values
+                        - advisory_data.loc[shifted_indices, "datetime"].values
+                    )
                 )
                 .astype("timedelta64[s]")
                 .astype(float)