keep index and sort through compute-durations cli

LAAC-LSCP · Feb 11, 2025 · a440fe1 · a440fe1
1 parent 171a39d
commit a440fe1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ChildProject/cmdline.py b/ChildProject/cmdline.py
@@ -726,12 +726,12 @@ def compute_durations(args):
 
     durations = project.compute_recordings_duration(profile=args.profile).dropna()
 
-    recordings = project.recordings.merge(
+    recordings = project.recordings.reset_index().merge(
         durations[durations["recording_filename"] != "NA"],
         how="left",
         left_on="recording_filename",
         right_on="recording_filename",
-    )
+    ).set_index('index')
     recordings["duration"].fillna(0, inplace=True)
     recordings["duration"] = recordings["duration"].astype("Int64")