From c45f241ecde602fefe23cc90c2b6df09a3b8d646 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Thu, 12 Oct 2023 08:03:31 -0700
Subject: [PATCH] Put back special case for python 3.11 - fix tensorrt

---
 .../actions/setup-binary-builds/action.yml    | 41 +++++++++++++++----
 1 file changed, 32 insertions(+), 9 deletions(-)

diff --git a/.github/actions/setup-binary-builds/action.yml b/.github/actions/setup-binary-builds/action.yml
index e4231e02d3..72d96fca78 100644
--- a/.github/actions/setup-binary-builds/action.yml
+++ b/.github/actions/setup-binary-builds/action.yml
@@ -109,16 +109,39 @@ runs:
           set -euxo pipefail
           CONDA_ENV="${RUNNER_TEMP}/conda_environment_${GITHUB_RUN_ID}"
 
-          conda create \
-            --yes --quiet \
-            --prefix "${CONDA_ENV}" \
-            "python=${PYTHON_VERSION}" \
-            cmake=3.22 \
-            conda-build=3.21 \
-            ninja=1.10 \
-            pkg-config=0.29 \
-            wheel=0.37
+          if [[ "${PYTHON_VERSION}" = "3.11" ]]; then
 
+            if [[ "${PACKAGE_TYPE:-}" == "conda" ]]; then
+              conda create \
+                --yes --quiet \
+                --prefix "${CONDA_ENV}" \
+                "python=3.8" \
+                cmake=3.22 \
+                conda-build=3.21 \
+                ninja=1.10 \
+                pkg-config=0.29 \
+                wheel=0.37
+            else
+              conda create \
+                --yes --quiet \
+                --prefix "${CONDA_ENV}" \
+                "python=${PYTHON_VERSION}" \
+                cmake=3.22 \
+                ninja=1.10 \
+                pkg-config=0.29 \
+                wheel=0.37
+            fi
+          else
+            conda create \
+              --yes --quiet \
+              --prefix "${CONDA_ENV}" \
+              "python=${PYTHON_VERSION}" \
+              cmake=3.22 \
+              conda-build=3.21 \
+              ninja=1.10 \
+              pkg-config=0.29 \
+              wheel=0.37
+          fi
 
           echo "CONDA_ENV=${CONDA_ENV}" >> "${GITHUB_ENV}"
           echo "CONDA_RUN=conda run -p ${CONDA_ENV}" >> "${GITHUB_ENV}"