From b3be67c656b27812d090d70edfdbe29d0875c93b Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Wed, 14 Feb 2018 16:27:17 -0500
Subject: [PATCH] fix
---
mhcflurry/train_allele_specific_models_command.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index 1c0e5021..b21360cf 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -497,7 +497,7 @@ def train_model(
print(alleles)
data = data.loc[data.allele.isin(alleles)]
assert len(data) >= pretrain_min_points, (len(data), pretrain_min_points)
- data = (data.allele == allele).astype(int).values
+ train_rounds = (data.allele == allele).astype(int).values
else:
train_rounds = None
data = data.loc[data.allele == allele]
--
GitLab