Fix rename error and add docs (#138)

* Push fix and documentation * Fix comment * Update glue/flagging_script_glue/flagging.py Co-authored-by: Jean Cochrane <[email protected]> --------- Co-authored-by: Jean Cochrane <[email protected]>
ccao-data · Dec 16, 2024 · 2523e85 · 2523e85
1 parent d7e1b9c
commit 2523e85
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/glue/flagging_script_glue/flagging.py b/glue/flagging_script_glue/flagging.py
@@ -572,9 +572,8 @@ def get_sale_counts(dups: pd.DataFrame) -> pd.DataFrame:
         df (pd.DataFrame): pandas dataframe
     """
     v_counts = (
-        dups.pin.value_counts()
-        .reset_index()
-        .rename(columns={"count": "sv_sale_dup_counts"})
+        dups.pin.value_counts().reset_index(name="count")
+        .rename(columns={"index": "pin"})
     )
 
     dups = pd.merge(dups, v_counts)