From 3445803552256f9e606df87509c9ed2a98a62f71 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 6 Dec 2019 15:33:26 -0500
Subject: [PATCH] fix
---
test/test_class1_presentation_neural_network.py | 1 -
test/test_class1_presentation_predictor.py | 4 ++--
2 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/test/test_class1_presentation_neural_network.py b/test/test_class1_presentation_neural_network.py
index 2d615b89..91799922 100644
--- a/test/test_class1_presentation_neural_network.py
+++ b/test/test_class1_presentation_neural_network.py
@@ -360,4 +360,3 @@ def test_real_data_multiallelic_refinement(max_epochs=10):
print(refine_allele, "final motif lysine-1 rate: ", final_motif.loc[1, "K"])
assert_greater(final_motif.loc[1, "K"], original_motif.loc[1, "K"])
-
\ No newline at end of file
diff --git a/test/test_class1_presentation_predictor.py b/test/test_class1_presentation_predictor.py
index 939de5c8..9b18f5d0 100644
--- a/test/test_class1_presentation_predictor.py
+++ b/test/test_class1_presentation_predictor.py
@@ -24,7 +24,7 @@ AFFINITY_PREDICTOR = None
def setup():
global AFFINITY_PREDICTOR
startup()
- PAN_ALLELE_PREDICTOR_NO_MASS_SPEC = Class1AffinityPredictor.load(
+ AFFINITY_PREDICTOR = Class1AffinityPredictor.load(
get_path("models_class1_pan", "models.no_mass_spec"),
optimization_level=0,
max_models=1)
@@ -32,7 +32,7 @@ def setup():
def teardown():
global AFFINITY_PREDICTOR
- PAN_ALLELE_PREDICTOR_NO_MASS_SPEC = None
+ AFFINITY_PREDICTOR = None
cleanup()
--
GitLab