From 0dc5caffa36698c082129d90e0098d3fd9a30f57 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Wed, 14 Feb 2018 16:26:20 -0500
Subject: [PATCH] fix
---
mhcflurry/train_allele_specific_models_command.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index d075167d..1c0e5021 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -475,12 +475,12 @@ def train_model(
pretrain_min_points = hyperparameters['train_data']['pretrain_min_points']
- full_data = GLOBAL_DATA["train_data"]
+ data = GLOBAL_DATA["train_data"]
subset = hyperparameters.get("train_data", {}).get("subset", "all")
if subset == "quantitative":
- data = full_data.loc[
- full_data.measurement_type == "quantitative"
+ data = data.loc[
+ data.measurement_type == "quantitative"
]
elif subset == "all":
pass
--
GitLab