From edcacd1e41185d73043202ca572d3b9659cc63f2 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Mon, 3 Feb 2025 15:05:44 -0600
Subject: [PATCH] avoid interactive prompts when updating from an
 environment.yaml (#727)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Working with @taureandyernv yesterday, we tried something like this to patch in a dependency at run time.

```shell
cat > ./environment.yml <<EOF
channels:
- dglteam/label/th23_cu121
dependencies:
- dgl
EOF

docker run \
  --rm \
  -v $(pwd)/environment.yml:/home/rapids/environment.yml \
  -it rapidsai/base:25.02a-cuda12.5-py3.12 \
  conda env export
```

Found that that prompts for confirmation:

```text
  Package           Version  Build    Channel                       Size
──────────────────────────────────────────────────────────────────────────
  Install:
──────────────────────────────────────────────────────────────────────────

  + dgl    2.4.0.th23.cu121  py312_0  dglteam/label/th23_cu121     295MB

  Summary:
  Install: 1 packages
  Total download: 295MB

──────────────────────────────────────────────────────────────────────────

Confirm changes: [Y/n]
```

But that prompt seems not to make it through to `mamba`.

And even if that did work, this interactive prompt prevents the use of the "mount in an `environment.yml`" approach in situations where you aren't running the container interactively... for example, where you want to use this image in a container in a Kubernetes Pod, and provide an `environment.yml` via a ConfigMap or file Secret.

This proposes skipping the confirmation and always running `mamba env update -y` when an `environment.yml` is provided.

## Notes for Reviewers

### How I tested this

Pulled an image built from this PR and confirmed that that install pattern described above works and does not prompt for interactive input.

```shell
cat > ./environment.yml <<EOF
channels:
- dglteam/label/th23_cu121
dependencies:
- dgl
EOF

docker run \
  --rm \
  -v $(pwd)/environment.yml:/home/rapids/environment.yml \
  -it rapidsai/staging:docker-727-25.02a-cuda12.5-py3.12 \
  conda env export
```

```text
...
  - dgl=2.4.0.th23.cu121=py312_0
...
```

Authors:
  - James Lamb (https://github.com/jameslamb)
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - Bradley Dice (https://github.com/bdice)

URL: https://github.com/rapidsai/docker/pull/727
---
 context/entrypoint.sh     | 2 +-
 context/test_notebooks.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/context/entrypoint.sh b/context/entrypoint.sh
index 82b9c8cf..7f27e5b8 100755
--- a/context/entrypoint.sh
+++ b/context/entrypoint.sh
@@ -10,7 +10,7 @@ EOF
 
 if [ -e "/home/rapids/environment.yml" ]; then
     echo "environment.yml found. Installing packages."
-    timeout ${CONDA_TIMEOUT:-600} mamba env update -n base -f /home/rapids/environment.yml || exit $?
+    timeout ${CONDA_TIMEOUT:-600} mamba env update -n base -y -f /home/rapids/environment.yml || exit $?
 fi
 
 if [ "$EXTRA_CONDA_PACKAGES" ]; then
diff --git a/context/test_notebooks.py b/context/test_notebooks.py
index a740c3af..e6ae167c 100755
--- a/context/test_notebooks.py
+++ b/context/test_notebooks.py
@@ -33,7 +33,7 @@
     'cuspatial/nyc_taxi_years_correlation.ipynb',
     # context on skip zipcodes: https://github.com/rapidsai/cuspatial/issues/1426
     'cuspatial/ZipCodes_Stops_PiP_cuSpatial.ipynb',
-    # context on this being skipped: https://github.com/rapidsai/docker/pull/728#issuecomment-2631146036
+    # context on this being skipped: https://github.com/rapidsai/docker/issues/726
     'cuspatial/trajectory_clustering.ipynb',
 ]