From 8c4055a0a6a31ccbd3b0a64670961c3598cfabc0 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sat, 21 Mar 2020 15:26:27 -0400
Subject: [PATCH] fix

---
 .travis.yml                                | 2 ++
 docs/commandline_tutorial.rst              | 2 +-
 test/test_class1_presentation_predictor.py | 4 ++--
 3 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index db7d5c23..6164353a 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,3 +1,4 @@
+dist: bionic
 language: python
 python:
   - "3.6"
@@ -17,6 +18,7 @@ before_install:
   # Useful for debugging any issues with conda
   - conda info -a
   - free -m
+  - make -v
 addons:
   apt:
     packages:
diff --git a/docs/commandline_tutorial.rst b/docs/commandline_tutorial.rst
index 874f6860..a2a6d22e 100644
--- a/docs/commandline_tutorial.rst
+++ b/docs/commandline_tutorial.rst
@@ -161,7 +161,7 @@ training data. The data we use for our released predictors can be downloaded wit
 It looks like this:
 
 .. command-output::
-    bzcat "$(mhcflurry-downloads path data_curated)/curated_training_data.no_mass_spec.csv.bz2" | head -n 3
+    bzcat "$(mhcflurry-downloads path data_curated)/curated_training_data.csv.bz2" | head -n 3
     :shell:
     :nostderr:
 
diff --git a/test/test_class1_presentation_predictor.py b/test/test_class1_presentation_predictor.py
index 35a9c6dc..e144c1fd 100644
--- a/test/test_class1_presentation_predictor.py
+++ b/test/test_class1_presentation_predictor.py
@@ -91,13 +91,13 @@ def test_basic():
             alleles=experiment_to_alleles,
             n_flanks=test_df.n_flank.values,
             c_flanks=test_df.c_flank.values,
-            verbose=2)
+            verbose=2).presentation_score.values
 
         test_df["prediction2"] = predictor.predict(
             peptides=test_df.peptide.values,
             sample_names=test_df.sample_id.values,
             alleles=experiment_to_alleles,
-            verbose=2)
+            verbose=2).presentation_score.values
 
     add_prediction_cols(test_df, predictor)
 
-- 
GitLab