diff --git a/ParaUtils/AWR/__init__.py b/ParaUtils/AWR/__init__.py index 6fddcb6..dcb1200 100644 --- a/ParaUtils/AWR/__init__.py +++ b/ParaUtils/AWR/__init__.py @@ -38,6 +38,6 @@ def find_AWR(performances,AWR=Current_AWR,identifier='SNC_ID'): on='Gender EventClass Stroke Distance'.split(), how='inner' ) - .groupby(identifier+'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