From 8946a57d6a92a9f8a8abdfb3dafffae1a15589de Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Wed, 29 Nov 2023 13:15:38 -0800
Subject: [PATCH] Apply classic solver to M1 builds

Add checkchannel where package is installed

test

test

test

test

test

test

test

test

test

test

test

test

Fix lint

typo
---
 .github/workflows/build_conda_linux.yml   | 24 ++++++++++++++++++++---
 .github/workflows/build_conda_macos.yml   | 22 +++++++++++++++++++--
 .github/workflows/build_conda_windows.yml | 24 ++++++++++++++++++++---
 3 files changed, 62 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/build_conda_linux.yml b/.github/workflows/build_conda_linux.yml
index 29b40ab1e1..e04bcb676c 100644
--- a/.github/workflows/build_conda_linux.yml
+++ b/.github/workflows/build_conda_linux.yml
@@ -180,14 +180,32 @@ jobs:
             CONSTRAINTS="cpuonly"
           fi
 
+          # From Conda documentation:
+          # https://docs.conda.io/projects/conda/en/latest/user-guide/concepts/channels.html#specifying-channels-when-installing-packages
+          # From the command line use --c
+          # You may specify multiple channels by passing the argument multiple times.
+          # Priority decreases from left to right - the first argument is higher priority than the second.
+          # From the command line use --override-channels to only search the specified channel(s),
+          # rather than any channels configured in .condarc.
+
           ${CONDA_RUN_SMOKE} conda install \
             --yes \
             --quiet \
+            -c "${CONDA_LOCAL_CHANNEL}" \
             -c pytorch-"${CHANNEL}" \
             -c nvidia \
-            -c "${CONDA_LOCAL_CHANNEL}" \
-            distr::"${PACKAGE_NAME}" \
-            ${CONSTRAINTS}
+            -c defaults \
+            "${PACKAGE_NAME}" \
+            ${CONSTRAINTS} \
+            --override-channels
+
+          # check for installed package channel
+          CONDA_PKG=$(${CONDA_RUN_SMOKE} conda list | grep "${PACKAGE_NAME}" | grep "${CONDA_LOCAL_CHANNEL}")
+          if [[ -z "${CONDA_PKG}" ]]; then
+            ${CONDA_RUN_SMOKE} conda list | grep "${PACKAGE_NAME}"
+            echo "${PACKAGE_NAME} is not installed from local channel"
+            exit 1
+          fi
 
           if [[ ! -f "${{ inputs.repository }}"/${SMOKE_TEST_SCRIPT} ]]; then
             echo "${{ inputs.repository }}/${SMOKE_TEST_SCRIPT} not found"
diff --git a/.github/workflows/build_conda_macos.yml b/.github/workflows/build_conda_macos.yml
index 29b99e5b47..a76f13dbd8 100644
--- a/.github/workflows/build_conda_macos.yml
+++ b/.github/workflows/build_conda_macos.yml
@@ -182,14 +182,32 @@ jobs:
           export CONDA_RUN_SMOKE="conda run -p ${CONDA_ENV_SMOKE}"
           export OLD_PATH=${PATH}
           export PATH="${CONDA_ENV_SMOKE}/bin:${PATH}"
-
           CONDA_LOCAL_CHANNEL="file://$(readlink -f ${{ inputs.repository }}/distr)"
+
+          # From Conda documentation:
+          # https://docs.conda.io/projects/conda/en/latest/user-guide/concepts/channels.html#specifying-channels-when-installing-packages
+          # From the command line use --c
+          # You may specify multiple channels by passing the argument multiple times.
+          # Priority decreases from left to right - the first argument is higher priority than the second.
+          # From the command line use --override-channels to only search the specified channel(s),
+          # rather than any channels configured in .condarc.
+
           ${CONDA_RUN_SMOKE} conda install \
             --quiet \
             --yes \
             -c "${CONDA_LOCAL_CHANNEL}" \
             -c pytorch-${CHANNEL} \
-            "${PACKAGE_NAME}"
+            -c defaults \
+            "${PACKAGE_NAME}" \
+            "--override-channels"
+
+          # check for installed package channel
+          CONDA_PKG=$(${CONDA_RUN_SMOKE} conda list | grep "${PACKAGE_NAME}" | grep "${CONDA_LOCAL_CHANNEL}")
+          if [[ -z "${CONDA_PKG}" ]]; then
+            ${CONDA_RUN_SMOKE} conda list | grep "${PACKAGE_NAME}"
+            echo "${PACKAGE_NAME} is not installed from local channel"
+            exit 1
+          fi
 
           if [[ ! -f "${{ inputs.repository }}"/${SMOKE_TEST_SCRIPT} ]]; then
             echo "${{ inputs.repository }}/${SMOKE_TEST_SCRIPT} not found"
diff --git a/.github/workflows/build_conda_windows.yml b/.github/workflows/build_conda_windows.yml
index 4ac0e5d12f..a349c6272e 100644
--- a/.github/workflows/build_conda_windows.yml
+++ b/.github/workflows/build_conda_windows.yml
@@ -200,14 +200,32 @@ jobs:
             CONSTRAINTS="cpuonly"
           fi
 
+          # From Conda documentation:
+          # https://docs.conda.io/projects/conda/en/latest/user-guide/concepts/channels.html#specifying-channels-when-installing-packages
+          # From the command line use --c
+          # You may specify multiple channels by passing the argument multiple times.
+          # Priority decreases from left to right - the first argument is higher priority than the second.
+          # From the command line use --override-channels to only search the specified channel(s),
+          # rather than any channels configured in .condarc.
+
           ${CONDA_RUN_SMOKE} conda install \
             --quiet \
             --yes \
+            -c "${CONDA_LOCAL_CHANNEL}" \
             -c pytorch-"${CHANNEL}" \
             -c nvidia \
-            -c "${CONDA_LOCAL_CHANNEL}" \
-            distr::"${PACKAGE_NAME}" \
-            ${CONSTRAINTS}
+            -c defaults \
+            "${PACKAGE_NAME}" \
+            ${CONSTRAINTS} \
+            --override-channels
+
+          # check for installed package channel
+          CONDA_PKG=$(${CONDA_RUN_SMOKE} conda list | grep "${PACKAGE_NAME}" | grep "${CONDA_LOCAL_CHANNEL}")
+          if [[ -z "${CONDA_PKG}" ]]; then
+            ${CONDA_RUN_SMOKE} conda list | grep "${PACKAGE_NAME}"
+            echo "${PACKAGE_NAME} is not installed from local channel"
+            exit 1
+          fi
 
           if [[ ! -f "${{ inputs.repository }}"/${SMOKE_TEST_SCRIPT} ]]; then
             echo "${{ inputs.repository }}/${SMOKE_TEST_SCRIPT} not found"