Skip to content

Commit e97ad19

Browse files
committed
make kernels set
1 parent 101ab77 commit e97ad19

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

template/template.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)