Skip to content
5 changes: 5 additions & 0 deletions .travis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,11 @@ before_install:
- conda info -a
- travis_retry conda create -n test $CONDA IPython pip nose pyzmq jsonschema
- source activate test
- if [[ $CONDA == python=3.3* ]]; then
pip install nbformat;
else
travis_retry conda install nbformat;
fi
- travis_retry pip install coveralls

install:
Expand Down
9 changes: 7 additions & 2 deletions pymatbridge/publish.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
import IPython.nbformat.v4 as nbformat
from IPython.nbformat import write as nbwrite
try:
import nbformat.v4 as nbformat
from nbformat import write as nbwrite
except ImportError:
import IPython.nbformat.v4 as nbformat
from IPython.nbformat import write as nbwrite

import numpy as np


Expand Down