fix

openvax · Jan 2, 2020 · 9d2ed0a · 9d2ed0a
1 parent 37e226b
commit 9d2ed0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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