From a8579ea3329a9f9509acb369c0eef28a779e3b53 Mon Sep 17 00:00:00 2001
From: Ayaz Salikhov <mathbunnyru@users.noreply.github.com>
Date: Sat, 30 Apr 2022 20:43:47 +0300
Subject: [PATCH] Fix choosing python version

---
 base-notebook/Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/base-notebook/Dockerfile b/base-notebook/Dockerfile
index 72b90a954e..376bb5fb13 100644
--- a/base-notebook/Dockerfile
+++ b/base-notebook/Dockerfile
@@ -84,7 +84,7 @@ RUN echo "auth requisite pam_deny.so" >> /etc/pam.d/su && \
 
 USER ${NB_UID}
 
-# Pin versions here, or set them to "default"
+# Pin python version here, or set it to "default"
 ARG PYTHON_VERSION=3.9
 
 # Setup work directory for backward-compatibility
@@ -115,6 +115,7 @@ RUN set -x && \
     tar -xvjf /tmp/micromamba.tar.bz2 --strip-components=1 bin/micromamba && \
     rm /tmp/micromamba.tar.bz2 && \
     PYTHON_SPECIFIER="python=${PYTHON_VERSION}" && \
+    if [[ "${PYTHON_VERSION}" == "default" ]]; then PYTHON_SPECIFIER="python"; fi && \
     if [ "${arch}" == "aarch64" ]; then \
         # Prevent libmamba from sporadically hanging on arm64 under QEMU
         # <https://github.com/mamba-org/mamba/issues/1611>