Fix rename

ocean-data-factory-sweden · Nov 22, 2023 · aaa80c2 · aaa80c2
1 parent 950559c
commit aaa80c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kso_utils/yolo_utils.py b/kso_utils/yolo_utils.py
@@ -1849,7 +1849,7 @@ def custom_class(x):
         inplace=True,
     )
     if len(names) > 0:
-        total_df.rename({"class_id": "species_name"}, inplace=True)
+        total_df.rename(columns={"class_id": "species_name"}, inplace=True)
     total_df = pd.merge(
         total_df,
         tracking_df[["tracker_id", "frame_no"]].groupby("tracker_id").first(),