From f3a5222eeb14cf14cce56e6a12aaea7a6e5e2db7 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 1 Oct 2019 17:18:21 -0400
Subject: [PATCH] fix
---
.../data_mass_spec_benchmark/run_mhcflurry.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
index 804cd531..5e2f6d3d 100644
--- a/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
+++ b/downloads-generation/data_mass_spec_benchmark/run_mhcflurry.py
@@ -6,7 +6,6 @@ import signal
import sys
import time
import traceback
-import collections
import math
from functools import partial
@@ -231,6 +230,10 @@ def run(argv=sys.argv[1:]):
def do_predictions(chunk_index, peptides, alleles, constant_data=None):
+ # This may run on the cluster in a way that misses all top level imports,
+ # so we have to re-import everything here.
+ import time
+
if constant_data is None:
constant_data = GLOBAL_DATA
--
GitLab