From 20e8e51e3350b802eec4d45b2613b98029752d11 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sun, 19 Feb 2017 14:37:39 -0500
Subject: [PATCH] small fixes

---
 mhcflurry/antigen_presentation/__init__.py                     | 3 ++-
 .../presentation_component_models/expression.py                | 2 +-
 .../fixed_affinity_predictions.py                              | 3 +++
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/mhcflurry/antigen_presentation/__init__.py b/mhcflurry/antigen_presentation/__init__.py
index 0b544c71..7406a57d 100644
--- a/mhcflurry/antigen_presentation/__init__.py
+++ b/mhcflurry/antigen_presentation/__init__.py
@@ -1,9 +1,10 @@
-from .presentation_model import PresentationModel
+from .presentation_model import PresentationModel, build_presentation_models
 from .percent_rank_transform import PercentRankTransform
 from . import presentation_component_models, decoy_strategies
 
 __all__ = [
     "PresentationModel",
+    "build_presentation_models",
     "PercentRankTransform",
     "presentation_component_models",
     "decoy_strategies",
diff --git a/mhcflurry/antigen_presentation/presentation_component_models/expression.py b/mhcflurry/antigen_presentation/presentation_component_models/expression.py
index 0c4b516a..22e1abb0 100644
--- a/mhcflurry/antigen_presentation/presentation_component_models/expression.py
+++ b/mhcflurry/antigen_presentation/presentation_component_models/expression.py
@@ -25,7 +25,7 @@ class Expression(PresentationComponentModel):
             group in expression_values.columns
             for group in experiment_to_expression_group.values())
 
-        assert_no_null(experiment_to_expression_group)
+        assert_no_null(expression_values)
 
         self.experiment_to_expression_group = experiment_to_expression_group
         self.expression_values = expression_values
diff --git a/mhcflurry/antigen_presentation/presentation_component_models/fixed_affinity_predictions.py b/mhcflurry/antigen_presentation/presentation_component_models/fixed_affinity_predictions.py
index 3a4a29b1..0680dde4 100644
--- a/mhcflurry/antigen_presentation/presentation_component_models/fixed_affinity_predictions.py
+++ b/mhcflurry/antigen_presentation/presentation_component_models/fixed_affinity_predictions.py
@@ -16,6 +16,9 @@ class FixedAffinityPredictions(PresentationComponentModel):
             - "value", "percentile_rank" (IC50 and percent rank)
             - peptide (string)
             - allele (string)
+
+    name : string
+        Used to name output columns and in debug messages
     """
 
     def __init__(
-- 
GitLab