From bb159a219eabd4a692246391d07f8682ef2f3bda Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Wed, 22 May 2024 21:26:44 -0700
Subject: [PATCH] Fix pip 22 bug, prefer vendor index URLs over PyPi

---
 setup.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index c8b2eb923..c3cee08d0 100644
--- a/setup.py
+++ b/setup.py
@@ -143,9 +143,9 @@ def dependencies(force_nightly: bool = False) -> List[str]:
     index_urls = [('https://pypi.org/simple', 'https://pypi.org/simple')]
     # prefer nvidia over AMD because AM5/iGPU systems will have a valid ROCm device
     if _is_nvidia():
-        index_urls += [nvidia_torch_index]
+        index_urls = [nvidia_torch_index] + index_urls
     elif _is_amd():
-        index_urls += [amd_torch_index]
+        index_urls = [amd_torch_index] + index_urls
         _dependencies += ["pytorch-triton-rocm"]
     else:
         index_urls += [cpu_torch_index]
@@ -169,7 +169,7 @@ def dependencies(force_nightly: bool = False) -> List[str]:
     except:
         try:
             # pip 22
-            finder = PackageFinder.create(LinkCollector(session, SearchScope([], index_urls)),  # type: ignore
+            finder = PackageFinder.create(LinkCollector(session, SearchScope([], index_urls_selected)),  # type: ignore
                                           SelectionPreferences(allow_yanked=False, prefer_binary=False,
                                                                allow_all_prereleases=True)
                                           , use_deprecated_html5lib=False)