@@ -5,7 +5,7 @@ def make_template(
55 kernels : list [str ] = ["python" , "r" , "javascript" , "deno" , "bash" , "java" ],
66 set_user_workdir : bool = False ,
77):
8- kernels = ["python" , "javascript" ] + kernels
8+ enabled_kernels = set ( ["python" , "javascript" ] + kernels )
99 # Start with base template
1010 template = (
1111 Template ()
@@ -46,11 +46,11 @@ def make_template(
4646 .pip_install ("--no-cache-dir -r requirements.txt" )
4747 )
4848
49- if "python" in kernels :
49+ if "python" in enabled_kernels :
5050 template = template .run_cmd ("ipython kernel install --name 'python3' --user" )
5151
5252 # Install R Kernel if requested
53- if "r" in kernels :
53+ if "r" in enabled_kernels :
5454 template = (
5555 template .run_cmd (
5656 [
@@ -69,14 +69,14 @@ def make_template(
6969 )
7070
7171 # Install JavaScript Kernel if requested
72- if "javascript" in kernels :
72+ if "javascript" in enabled_kernels :
7373 template = template .npm_install (
7474 "--unsafe-perm git+https://github.com/e2b-dev/ijavascript.git" ,
7575 g = True ,
7676 ).run_cmd ("ijsinstall --install=global" )
7777
7878 # Install Deno Kernel if requested
79- if "deno" in kernels :
79+ if "deno" in enabled_kernels :
8080 template = template .run_cmd (
8181 [
8282 "curl -fsSL https://deno.land/install.sh | sh -s ${DENO_VERSION}" ,
@@ -86,13 +86,13 @@ def make_template(
8686 ).copy ("deno.json" , ".local/share/jupyter/kernels/deno/kernel.json" )
8787
8888 # Install Bash Kernel if requested
89- if "bash" in kernels :
89+ if "bash" in enabled_kernels :
9090 template = template .pip_install ("bash_kernel" ).run_cmd (
9191 "python -m bash_kernel.install"
9292 )
9393
9494 # Install Java and Java Kernel if requested
95- if "java" in kernels :
95+ if "java" in enabled_kernels :
9696 template = template .run_cmd (
9797 [
9898 "mkdir -p /usr/lib/jvm" ,
0 commit comments