diff --git a/jass_preprocessing/compute_score.py b/jass_preprocessing/compute_score.py index 1a1217ac39acadc3552c68134cc57c6d86317512..fe7425c0857d34f9617bbf67b2b816881ba80e3a 100644 --- a/jass_preprocessing/compute_score.py +++ b/jass_preprocessing/compute_score.py @@ -25,13 +25,9 @@ def compute_z_score(mgwas): print(mgwas.columns) mgwas["computed_z"] = np.sqrt(ss.chi2.isf(mgwas['pval'], 1)) - if 'z' in mgwas.columns: - sign_vect = np.sign(mgwas.z) - - mgwas.loc[mgwas["computed_z"].isin([np.inf]), 'computed_z'] = mgwas["z"] - elif 'beta_or_Z' in mgwas.columns: - sign_vect = np.sign(mgwas.beta) + if 'beta_or_Z' in mgwas.columns: + sign_vect = np.sign(mgwas.beta_or_Z) mgwas["computed_z"] = mgwas["computed_z"].replace(np.inf, 37.537836095576054) else: if "OR" in mgwas.columns: