From cd4e49c138e9722fecd43d2878d1bf6d925a5753 Mon Sep 17 00:00:00 2001
From: James Lamb <jaylamb20@gmail.com>
Date: Mon, 2 Dec 2024 13:02:03 -0600
Subject: [PATCH] ci-conda: change spelling of python dependency (#214)

---
 ci-conda.Dockerfile      | 6 ++++--
 ci/compute-build-args.sh | 6 ++++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/ci-conda.Dockerfile b/ci-conda.Dockerfile
index bc6b2dc..0d76e9b 100644
--- a/ci-conda.Dockerfile
+++ b/ci-conda.Dockerfile
@@ -8,6 +8,7 @@ FROM nvidia/cuda:${CUDA_VER}-base-${LINUX_VER} AS miniforge-cuda
 
 ARG LINUX_VER
 ARG PYTHON_VER
+ARG PYTHON_VER_UPPER_BOUND
 ARG DEBIAN_FRONTEND=noninteractive
 ENV PATH=/opt/conda/bin:$PATH
 ENV PYTHON_VERSION=${PYTHON_VER}
@@ -33,7 +34,7 @@ umask 002
 # an older conda with newer packages still works well
 conda update --all -y -n base
 # install expected Python version
-conda install -y -n base "python~=${PYTHON_VERSION}.0=*_cpython"
+conda install -y -n base "python>=${PYTHON_VERSION},<${PYTHON_VER_UPPER_BOUND}=*_cpython"
 conda update --all -y -n base
 if [[ "$LINUX_VER" == "rockylinux"* ]]; then
   yum install -y findutils
@@ -93,6 +94,7 @@ ARG TARGETPLATFORM=notset
 ARG CUDA_VER=notset
 ARG LINUX_VER=notset
 ARG PYTHON_VER=notset
+ARG PYTHON_VER_UPPER_BOUND=notset
 
 ARG DEBIAN_FRONTEND
 
@@ -212,7 +214,7 @@ rapids-mamba-retry install -y \
   git \
   jq \
   packaging \
-  "python=${PYTHON_VERSION}.*=*_cpython" \
+  "python>=${PYTHON_VERSION},<${PYTHON_VER_UPPER_BOUND}=*_cpython" \
   "rapids-dependency-file-generator==1.*"
 conda clean -aipty
 EOF
diff --git a/ci/compute-build-args.sh b/ci/compute-build-args.sh
index 7a1087d..e542183 100755
--- a/ci/compute-build-args.sh
+++ b/ci/compute-build-args.sh
@@ -17,10 +17,16 @@ if [[
   MANYLINUX_VER="manylinux_2_31"
 fi
 
+# compute upper bound, e.g. "3.11 -> 3.12.0a0"
+PYTHON_VER_MAJOR="${PYTHON_VER%%.*}"
+PYTHON_VER_MINOR="${PYTHON_VER#*.}"
+PYTHON_VER_UPPER_BOUND="${PYTHON_VER_MAJOR}.$(( PYTHON_VER_MINOR + 1)).0a0"
+
 ARGS="
 CUDA_VER: ${CUDA_VER}
 LINUX_VER: ${LINUX_VER}
 PYTHON_VER: ${PYTHON_VER}
+PYTHON_VER_UPPER_BOUND: ${PYTHON_VER_UPPER_BOUND}
 CPU_ARCH: ${ARCH}
 REAL_ARCH: $(arch)
 MANYLINUX_VER: ${MANYLINUX_VER}