diff --git a/ParaUtils/AWR/__init__.py b/ParaUtils/AWR/__init__.py index e03e54e..6fddcb6 100644 --- a/ParaUtils/AWR/__init__.py +++ b/ParaUtils/AWR/__init__.py @@ -24,7 +24,7 @@ Current_AWR = ( pd.read_excel(module_dir / '2025 Fall Performance AWR - 2025-10-07.xlsx') ) -def find_AWR(performances,AWR=Current_AWR): +def find_AWR(performances,AWR=Current_AWR,identifier='SNC_ID'): return ( performances .astype({'SportClass':str}) @@ -38,6 +38,6 @@ def find_AWR(performances,AWR=Current_AWR): on='Gender EventClass Stroke Distance'.split(), how='inner' ) - .groupby('Gender EventClass Stroke Distance'.split(),group_keys=False) + .groupby(identifier+'Gender EventClass Stroke Distance'.split(),group_keys=False) .apply(lambda df: df.assign(AWR=1+len(df[df.AWR < df.Seconds])).head(1)) ) \ No newline at end of file