fix bug in annotation name pruning when we have >9 annotations

RobinM-code · Apr 26, 2022 · 8c4927f · 8c4927f
1 parent 28e7f2b
commit 8c4927f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/polyfun.py b/polyfun.py
@@ -213,7 +213,7 @@ def run_ldsc(self, args, use_ridge, nn, keep_large, evenodd_split, n_blocks=2):
         chisq = chisq[ii].reshape((n_snp, 1))
 
         #Run S-LDSC
-        self.ref_ld_cnames = [c for c in ref_ld_cnames.str[:-2] if c not in SNP_COLUMNS]
+        self.ref_ld_cnames = [c[:c.rfind('_')] for c in ref_ld_cnames if c not in SNP_COLUMNS]
         hsqhat = regressions.Hsq(chisq, 
             ref_ld,
             s(df_sumstats[w_ld_cname]),