Minor fix

stella-bourdin · Apr 7, 2021 · db9878e · db9878e
1 parent 1d62eed
commit db9878e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dynamicopy/cyclones.py b/dynamicopy/cyclones.py
@@ -19,6 +19,7 @@ def load_ibtracs(
         .rename(columns={"usa_sshs": "sshs", "sid": "track_id"})
         .drop(columns="season")
     )
+    tracks['basin'] = tracks.basin.replace('EP', 'ENP').replace('WP', 'WNP')
     tracks["hemisphere"] = np.where(tracks.lat > 0, "N", "S")
     tracks = add_season(tracks)
     return tracks