fix hadnling of non-human organisms

omerwe · May 9, 2024 · 6e3c5d9 · 6e3c5d9
1 parent 05fcd49
commit 6e3c5d9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ldsc_polyfun/sumstats.py b/ldsc_polyfun/sumstats.py
@@ -248,6 +248,7 @@ def _merge_and_log(ld, sumstats, noun, log):
 
 
 def _read_ld_sumstats(args, log, fh, alleles=True, dropna=True):
+    global _N_CHR
     _N_CHR = args.num_chr
     sumstats = _read_sumstats(args, log, fh, alleles=alleles, dropna=dropna)
     ref_ld = _read_ref_ld(args, log)
@@ -283,6 +284,7 @@ def _read_ld_sumstats(args, log, fh, alleles=True, dropna=True):
 def estimate_h2(args, log):
     '''Estimate h2 and partitioned h2.'''
     args = copy.deepcopy(args)
+    global _N_CHR
     _N_CHR = args.num_chr
     if args.samp_prev is not None and args.pop_prev is not None:
         args.samp_prev, args.pop_prev = list(map(