dask widget in notebooks not working

After installing the toolbox in a proposal, and trying to display a dask array object in a notebook I get the following error:

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
/gpfs/exfel/exp/SCS/202022/p002712/usr/Software/envs/toolbox_p002712/lib/python3.7/site-packages/dask/widgets/__init__.py in <module>
      1 try:
----> 2     from .widgets import FILTERS, TEMPLATE_PATHS, get_environment, get_template
      3 

/gpfs/exfel/exp/SCS/202022/p002712/usr/Software/envs/toolbox_p002712/lib/python3.7/site-packages/dask/widgets/widgets.py in <module>
      4 
----> 5 from jinja2 import Environment, FileSystemLoader, Template
      6 from jinja2.exceptions import TemplateNotFound

ModuleNotFoundError: No module named 'jinja2'

The above exception was the direct cause of the following exception:

ImportError                               Traceback (most recent call last)
/gpfs/exfel/exp/SCS/202022/p002712/usr/Software/envs/toolbox_p002712/lib/python3.7/site-packages/IPython/core/formatters.py in __call__(self, obj)
    343             method = get_real_method(obj, self.print_method)
    344             if method is not None:
--> 345                 return method()
    346             return None
    347         else:

/gpfs/exfel/exp/SCS/202022/p002712/usr/Software/envs/toolbox_p002712/lib/python3.7/site-packages/dask/array/core.py in _repr_html_(self)
   1567             cbytes = "unknown"
   1568 
-> 1569         return get_template("array.html.j2").render(
   1570             array=self,
   1571             grid=grid,

/gpfs/exfel/exp/SCS/202022/p002712/usr/Software/envs/toolbox_p002712/lib/python3.7/site-packages/dask/widgets/__init__.py in get_template(name)
     17 
     18     def get_template(name: str):
---> 19         raise ImportError(msg) from exception

ImportError: Dask diagnostics requirements are not installed.

Please either conda or pip install as follows:

  conda install dask                     # either conda install
  python -m pip install "dask[diagnostics]" --upgrade  # or python -m pip install

dask.array<truediv, shape=(128, 256), dtype=float64, chunksize=(128, 256), chunktype=numpy.ndarray>