From 5bb1f547563cc3e51f8b607ca806f05312f5f227 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Wed, 16 Nov 2016 16:32:12 -0500
Subject: [PATCH] Pin keras version
Pin keras version. Also add note on KERAS_BACKEND to README.md.
Closes #67. Closes #65.
---
README.md | 11 ++++++++++-
requirements.txt | 2 +-
setup.py | 2 +-
3 files changed, 12 insertions(+), 3 deletions(-)
diff --git a/README.md b/README.md
index 1227198b..62efbf61 100644
--- a/README.md
+++ b/README.md
@@ -13,7 +13,16 @@ The MHCflurry predictors are implemented in Python using [keras](https://keras.i
## Setup
-Install the package:
+To configure keras, the neural network library used by MHCflurry, you'll need to set an environment variable in your shell:
+
+```
+export KERAS_BACKEND=theano
+```
+
+If you're familiar with keras, you may also try using the tensorflow backend. MHCflurry is currently tested using theano, however.
+
+
+Now install the package:
```
pip install mhcflurry
diff --git a/requirements.txt b/requirements.txt
index 0893a3a0..78a52a93 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ numpy>= 1.11
pandas>=0.13.1
appdirs
theano>=0.8.2
-keras
+keras==1.1.0
fancyimpute>=0.0.12
scikit-learn
h5py
diff --git a/setup.py b/setup.py
index 796b1ccb..07adad40 100644
--- a/setup.py
+++ b/setup.py
@@ -76,7 +76,7 @@ if __name__ == '__main__':
'pandas>=0.13.1',
'appdirs',
'theano>=0.8.2',
- 'keras',
+ 'keras==1.1.0',
'fancyimpute>=0.0.12',
'scikit-learn',
'h5py',
--
GitLab