From 34c6c7b69152b8017728cb56f7743fa9a1c0fb21 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 1 Oct 2019 17:04:20 -0400
Subject: [PATCH] fix
---
.../data_mass_spec_benchmark/run_mhcflurry.py | 24 +++++++------------
1 file changed, 8 insertions(+), 16 deletions(-)
diff --git a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
index 1b4253ae..804cd531 100644
--- a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
+++ b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
@@ -230,22 +230,14 @@ def run(argv=sys.argv[1:]):
prediction_time / 60.0))
-def do_predictions(chunk_index, peptides, alleles, constant_data=GLOBAL_DATA):
- return predict_for_allele(
- chunk_index,
- peptides,
- alleles,
- predictor=constant_data['predictor'],
- **constant_data["args"])
-
-
-def predict_for_allele(
- chunk_index,
- peptides,
- alleles,
- predictor,
- verbose=False,
- model_kwargs={}):
+def do_predictions(chunk_index, peptides, alleles, constant_data=None):
+ if constant_data is None:
+ constant_data = GLOBAL_DATA
+
+ predictor = constant_data['predictor']
+ verbose = constant_data['args'].get("verbose", False)
+ model_kwargs = constant_data['args'].get("model_kwargs", {})
+
predictor.optimize(warn=False) # since we loaded with optimization_level=0
start = time.time()
results = {}
--
GitLab