Use python 3.12 for nemo

Author: pantherman594

src/nemo_jupyter/.devcontainer.json

@@ -25,6 +25,7 @@
     },
     "ghcr.io/dhoeric/features/google-cloud-cli@sha256:fa5d894718825c5ad8009ac8f2c9f0cea3d1661eb108a9d465cba9f3fc48965f": {},
     "./.devcontainer/features/workbench-tools": {
+      "libPythonVersion": "3.12", // Must match python version in nemo image
       "cloud": "${templateOption:cloud}",
       "username": "jupyter",
       "userHomeDir": "/home/jupyter"

src/nemo_jupyter/Dockerfile

@@ -17,9 +17,15 @@ RUN useradd --uid ${NB_UID} --gid ${NB_GID} --create-home --home-dir ${WORKDIR}
     # Fix ownership for common dirs
     && mkdir -p /workspace \
     && chown -R ${NB_UID}:${NB_GID} ${WORKDIR} /workspace /tmp \
-    && chown -R ${NB_UID}:${NB_GID} /opt/conda || true \
-    # Add alias for weightsbiases
-    && printf 'alias weightsbiases="/usr/local/bin/wb"\nalias wb="/usr/bin/wb"\n' >> ${WORKDIR}/.bashrc
+    && chown -R ${NB_UID}:${NB_GID} /opt/conda || true
+
+# Add shell functions to override wb commands (functions work in non-interactive shells, unlike aliases)
+RUN <<EOF cat >> ${WORKDIR}/.bashrc
+weightsbiases() { /usr/local/bin/wb "\$@"; }
+wb() { /usr/bin/wb "\$@"; }
+wandb() { /opt/venv/bin/wb "\$@"; }
+export -f weightsbiases wb wandb
+EOF
 
 # Environment and working directory
 ENV HOME=${WORKDIR}

src/nemo_jupyter_aou/.devcontainer.json

@@ -26,6 +26,7 @@
     },
     "ghcr.io/dhoeric/features/google-cloud-cli@sha256:fa5d894718825c5ad8009ac8f2c9f0cea3d1661eb108a9d465cba9f3fc48965f": {},
     "./.devcontainer/features/workbench-tools": {
+      "libPythonVersion": "3.12", // Must match python version in nemo image
       "cloud": "${templateOption:cloud}",
       "username": "jupyter",
       "userHomeDir": "/home/jupyter"