From 9d2ed0aeedd16794b5d556b9b119581ce75d385b Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Thu, 2 Jan 2020 07:54:26 -0500
Subject: [PATCH] fix

---
 mhcflurry/select_pan_allele_models_command.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mhcflurry/select_pan_allele_models_command.py b/mhcflurry/select_pan_allele_models_command.py
index 9167ac13..92a01702 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
-- 
GitLab