From b7087aa89b79d01d9ecd3fd3a1a06a20b141ffbc Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 11 Oct 2019 11:15:54 -0400
Subject: [PATCH] fix
---
mhcflurry/train_pan_allele_models_command.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mhcflurry/train_pan_allele_models_command.py b/mhcflurry/train_pan_allele_models_command.py
index 2334850b..a608f8aa 100644
--- a/mhcflurry/train_pan_allele_models_command.py
+++ b/mhcflurry/train_pan_allele_models_command.py
@@ -309,7 +309,7 @@ def initialize_training(args):
pprint.pprint(hyperparameters_lst)
allele_sequences = pandas.read_csv(
- args.allele_sequences, index_col=0).sequence
+ args.allele_sequences, index_col=0).iloc[:,0]
df = pandas.read_csv(args.data)
print("Loaded training data: %s" % (str(df.shape)))
--
GitLab