From df52a69ac723a5884ad9b79411b39a7aec958cf7 Mon Sep 17 00:00:00 2001
From: Benjamin Wingfield <bwingfield@ebi.ac.uk>
Date: Wed, 21 Feb 2024 11:34:31 +0000
Subject: [PATCH] drop onlySUM parameter

---
 pgscatalog_utils/ancestry/read.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/pgscatalog_utils/ancestry/read.py b/pgscatalog_utils/ancestry/read.py
index ecbaf15..2db224e 100644
--- a/pgscatalog_utils/ancestry/read.py
+++ b/pgscatalog_utils/ancestry/read.py
@@ -73,11 +73,10 @@ def extract_ref_psam_cols(
     return pd.merge(df_target, psam[keepcols], left_index=True, right_index=True)
 
 
-def read_pgs(loc_aggscore, onlySUM: bool):
+def read_pgs(loc_aggscore):
     """
     Function to read the output of aggreagte_scores
     :param loc_aggscore: path to aggregated scores output
-    :param onlySUM: whether to return only _SUM columns (e.g. not _AVG)
     :return:
     """
     logger.debug("Reading aggregated score data: {}".format(loc_aggscore))
@@ -91,8 +90,4 @@ def read_pgs(loc_aggscore, onlySUM: bool):
     # join column levels ({PGS}_{VALUE})
     df.columns = [f"{j}_{i}" for i, j in df.columns]
 
-    if onlySUM:
-        df = df[[x for x in df.columns if x.endswith("_SUM")]]
-        rn = [x.rstrip("_SUM") for x in df.columns]
-        df.columns = rn
     return df