From 7acb41d57c92e19a4d94fe85e82b3bad05cd7782 Mon Sep 17 00:00:00 2001
From: George Necula <gcnecula@gmail.com>
Date: Mon, 11 May 2020 14:37:17 +0300
Subject: [PATCH] Fixed imports for readthedocs (#3033)

---
 docs/developer.rst    | 5 +++--
 docs/requirements.txt | 2 ++
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/docs/developer.rst b/docs/developer.rst
index 1ca7e9fc7..289e59179 100644
--- a/docs/developer.rst
+++ b/docs/developer.rst
@@ -188,10 +188,11 @@ I saw in the Readthedocs logs::
     cd jax
     git checkout --force origin/test-docs
     git clean -d -f -f
-    
+    workon jax-docs
+
     python -m pip install --upgrade --no-cache-dir pip
     python -m pip install --upgrade --no-cache-dir -I Pygments==2.3.1 setuptools==41.0.1 docutils==0.14 mock==1.0.1 pillow==5.4.1 alabaster>=0.7,<0.8,!=0.7.5 commonmark==0.8.1 recommonmark==0.5.0 'sphinx<2' 'sphinx-rtd-theme<0.5' 'readthedocs-sphinx-ext<1.1'
     python -m pip install --exists-action=w --no-cache-dir -r docs/requirements.txt
-    
+    cd docs
     python `which sphinx-build` -T -E -b html -d _build/doctrees-readthedocs -D language=en . _build/html
 
diff --git a/docs/requirements.txt b/docs/requirements.txt
index c7f76dce0..7d716c431 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -4,6 +4,8 @@ jaxlib
 ipykernel
 nbsphinx
 sphinx-autodoc-typehints
+# For host_callback.py
+msgpack
 # The next packages are for notebooks
 matplotlib
 sklearn