From de4615fa75c52c4a6827a6c89f423d0fadb6bcf2 Mon Sep 17 00:00:00 2001
From: Alex Rubinsteyn <alex.rubinsteyn@gmail.com>
Date: Tue, 19 Apr 2016 14:39:37 -0400
Subject: [PATCH] train very limited models on travis

---
 .travis.yml | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/.travis.yml b/.travis.yml
index 4a71970a..a6b3b506 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -36,6 +36,14 @@ install:
   - pip install .
   - pip install coveralls
 script:
+  # only installing A0101 and A0201 for testing purposes and with very limited
+  # training
+  - >
+      mhcflurry-train-class1-allele-specific-models.py
+      --alleles HLA-A0101 HLA-A0101
+      --embedding-size 4
+      --hidden-layer-size 4
+      --training-epochs 5
   # run tests
   - nosetests test --with-coverage --cover-package=mhcflurry  && ./lint.sh
 after_success:
-- 
GitLab