dep: added pyehm as dependency

Dockerfile

@@ -2,6 +2,6 @@ from jupyter/scipy-notebook
 
 # RUN apt update && apt install -y python3-pip
 
-RUN pip3 install jax jaxlib
+RUN pip3 install jax jaxlib pyehm
 
 ENTRYPOINT /bin/bash

Pipfile

@@ -14,6 +14,7 @@ jupyterlab = "*"
 ipywidgets = "*"
 jax = "*"
 jaxlib = "*"
+pyehm = "*"
 
 [requires]
 python_version = "3.8"