From 612fe5c831879066d0b6807f8e99dfb390cbb50f Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Thu, 8 Feb 2018 18:45:52 -0500
Subject: [PATCH] Switch back to mean instead of robust_mean default
---
mhcflurry/class1_affinity_predictor.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/mhcflurry/class1_affinity_predictor.py b/mhcflurry/class1_affinity_predictor.py
index 0b90d8c3..5cf8bf5f 100644
--- a/mhcflurry/class1_affinity_predictor.py
+++ b/mhcflurry/class1_affinity_predictor.py
@@ -707,7 +707,7 @@ class Class1AffinityPredictor(object):
throw=True,
include_individual_model_predictions=False,
include_percentile_ranks=True,
- centrality_measure="robust_mean"):
+ centrality_measure="mean"):
"""
Predict nM binding affinities. Gives more detailed output than `predict`
method, including 5-95% prediction intervals.
@@ -748,6 +748,8 @@ class Class1AffinityPredictor(object):
raise TypeError("peptides must be a list or array, not a string")
if isinstance(alleles, string_types):
raise TypeError("alleles must be a list or array, not a string")
+ if allele is None and alleles is None:
+ raise ValueError("Must specify 'allele' or 'alleles'.")
if allele is not None:
if alleles is not None:
raise ValueError("Specify exactly one of allele or alleles")
--
GitLab