From 581233be011bbfab6bd2583e65c6c4cbb928e5bb Mon Sep 17 00:00:00 2001
From: David Fairbrother <DavidFair@users.noreply.github.com>
Date: Thu, 17 Aug 2023 11:08:03 +0100
Subject: [PATCH] Get tensorboard working again

The new upstream image does not include the jupyter-server-proxy
package, which is required to embed Tensorboard with the env var.

Based on the upstream approach, install this using mamba then fix-up any
perms we might have changed.
---
 jupyter-tensorflow-notebook/Dockerfile | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/jupyter-tensorflow-notebook/Dockerfile b/jupyter-tensorflow-notebook/Dockerfile
index 7c0f49b..88404ff 100644
--- a/jupyter-tensorflow-notebook/Dockerfile
+++ b/jupyter-tensorflow-notebook/Dockerfile
@@ -15,8 +15,15 @@ RUN apt-get update && \
 
 USER ${NB_UID}
 
+# Required for Tensorboard on a remote container
+RUN mamba install --yes \
+    'jupyter-server-proxy' && \
+    mamba clean --all -f -y && \
+    fix-permissions "${CONDA_DIR}" && \
+    fix-permissions "/home/${NB_USER}"
+
+# SciML requirements
 RUN pip install --user --no-cache-dir \
-    # SciML requirements
     gym==0.25.2 \
     pygame && \
     fix-permissions "${CONDA_DIR}" && \