fix

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

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()