diff --git a/mhcflurry/select_pan_allele_models_command.py b/mhcflurry/select_pan_allele_models_command.py index 9167ac13e8b96a64911369d96c17e63f006b4932..92a0170238277e681ce00fa4f26d61d503e31c46 100644 --- a/mhcflurry/select_pan_allele_models_command.py +++ b/mhcflurry/select_pan_allele_models_command.py @@ -172,7 +172,7 @@ def run(argv=sys.argv[1:]): print(df[fold_cols].mean()) # Allele names in data are assumed to be already normalized. - df = df.loc[df.allele.isin(alleles)].dropna() + df = df.loc[df.allele.isin(alleles)] print("Subselected to supported alleles: %s" % str(df.shape)) metadata_dfs["model_selection_data"] = df