From e8f862a2bc32b75d9edcaed5dcfccc3e7eb32bbe Mon Sep 17 00:00:00 2001
From: hotwa <pylyzeng@gmail.com>
Date: Sat, 25 May 2024 18:30:00 +0800
Subject: [PATCH] update mojo

---
 spawnerdockerfile/Dockerfile.base-notebook | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/spawnerdockerfile/Dockerfile.base-notebook b/spawnerdockerfile/Dockerfile.base-notebook
index 9ac8703..f9cb670 100755
--- a/spawnerdockerfile/Dockerfile.base-notebook
+++ b/spawnerdockerfile/Dockerfile.base-notebook
@@ -138,12 +138,17 @@ USER ${NB_UID}
 ARG MODULAR_HOME="/home/${CREATE_USER}/.modular"
 ENV MODULAR_HOME=$MODULAR_HOME
 ENV PATH="$MODULAR_HOME/pkg/packages.modular.com_mojo/bin:/home/${CREATE_USER}/.local/bin:$PATH"
+ARG HTTP_PROXY=""
+ARG HTTPS_PROXY=""
+ARG NO_PROXY="localhost,127.0.0.1"
+ENV HTTP_PROXY=$HTTP_PROXY
+ENV HTTPS_PROXY=$HTTPS_PROXY
 RUN <<EOT
 #!/bin/bash
-curl https://get.modular.com | sh - 
-mamba create -n mojo python=3.10 -c conda-forge -y 
-mamba activate mojo
-modular install mojo
+curl --retry 5 https://get.modular.com | sh - 
+mamba create -n mojo python=3.10 ipykernel ipython nglview -c conda-forge -y 
+mamba activate mojo && modular install mojo
+mamba run -n mojo python -m ipykernel install --user --name="mojo" --display-name="mojo_env"
 EOT
 
 # Install Rust