From d2fcd7d7985a3afe8315398c51da99f998fa8508 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sun, 23 Sep 2018 15:08:24 -0400
Subject: [PATCH] fix
---
mhcflurry/allele_encoding.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/mhcflurry/allele_encoding.py b/mhcflurry/allele_encoding.py
index 61140b46..e7d47cf0 100644
--- a/mhcflurry/allele_encoding.py
+++ b/mhcflurry/allele_encoding.py
@@ -1,3 +1,5 @@
+from six import callable
+
import pandas
from . import amino_acid
@@ -72,7 +74,10 @@ class AlleleEncoding(object):
"allele_representations",
encoding_name)
if cache_key not in self.encoding_cache:
- if ":" in encoding_name:
+ if callable(encoding_name):
+ vector_encoded = encoding_name(self)
+ assert len(vector_encoded)== len(self.allele_to_sequence)
+ elif ":" in encoding_name:
# Apply transform
(transform_name, rest) = encoding_name.split(":", 2)
preliminary_encoded = self.allele_representations(rest)
--
GitLab