ci: simplify local wheel selection

Author: kkraus14

.github/workflows/test-wheel-linux.yml

@@ -390,26 +390,16 @@ jobs:
           if ${{ inputs.test-bindings || inputs.test-core }}; then
             dependency_args=(--no-deps)
           else
-            pathfinder_wheels=(./cuda_pathfinder/cuda_pathfinder-*.whl)
-            bindings_wheels=("${CUDA_BINDINGS_ARTIFACTS_DIR}"/cuda_bindings-*.whl)
-            test "${#pathfinder_wheels[@]}" -eq 1
-            test "${#bindings_wheels[@]}" -eq 1
-            test -f "${pathfinder_wheels[0]}"
-            test -f "${bindings_wheels[0]}"
-            if [[ "${{ matrix.LOCAL_CTK }}" == 1 ]]; then
-              dependency_args=("${pathfinder_wheels[0]}" "${bindings_wheels[0]}")
-            else
-              dependency_args=("${pathfinder_wheels[0]}" "${bindings_wheels[0]}[all]")
-            fi
+            dependency_args=(
+              ./cuda_pathfinder/cuda_pathfinder-*.whl
+              "${CUDA_BINDINGS_ARTIFACTS_DIR}"/cuda_bindings-*.whl
+            )
           fi
-          python_wheels=(cuda_python*.whl)
-          test "${#python_wheels[@]}" -eq 1
-          test -f "${python_wheels[0]}"
-          if [[ "${{ matrix.LOCAL_CTK }}" == 1 ]]; then
-            pip install --only-binary=:all: "${dependency_args[@]}" "${python_wheels[0]}"
-          else
-            pip install --only-binary=:all: "${dependency_args[@]}" "${python_wheels[0]}[all]"
+          python_requirements=(cuda_python*.whl)
+          if [[ "${{ matrix.LOCAL_CTK }}" != 1 ]]; then
+            python_requirements=("${python_requirements[@]/%/[all]}")
           fi
+          pip install --only-binary=:all: "${dependency_args[@]}" "${python_requirements[@]}"
 
       - name: Install cuda.pathfinder extra wheels for testing
         if: ${{ inputs.test-mode == 'standard' && inputs.test-pathfinder }}

.github/workflows/test-wheel-windows.yml

@@ -360,32 +360,16 @@ jobs:
           if ('${{ inputs.test-bindings || inputs.test-core }}' -eq 'true') {
             $dependencyArgs = @('--no-deps')
           } else {
-            $pathfinderWheels = @(Get-ChildItem -Path ./cuda_pathfinder -Filter cuda_pathfinder-*.whl -File)
-            $bindingsWheels = @(Get-ChildItem -Path $env:CUDA_BINDINGS_ARTIFACTS_DIR -Filter cuda_bindings-*.whl -File)
-            if ($pathfinderWheels.Count -ne 1) {
-              throw "Expected exactly one cuda-pathfinder wheel, found $($pathfinderWheels.Count)"
-            }
-            if ($bindingsWheels.Count -ne 1) {
-              throw "Expected exactly one cuda-bindings wheel, found $($bindingsWheels.Count)"
-            }
-            $pathfinderWheel = $pathfinderWheels[0].FullName
-            $bindingsWheel = $bindingsWheels[0].FullName
-            if ('${{ matrix.LOCAL_CTK }}' -eq '1') {
-              $dependencyArgs = @($pathfinderWheel, $bindingsWheel)
-            } else {
-              $dependencyArgs = @($pathfinderWheel, "${bindingsWheel}[all]")
-            }
-          }
-          $pythonWheels = @(Get-ChildItem -Filter cuda_python*.whl -File)
-          if ($pythonWheels.Count -ne 1) {
-            throw "Expected exactly one cuda-python wheel, found $($pythonWheels.Count)"
+            $dependencyArgs = @(
+              (Get-Item ./cuda_pathfinder/cuda_pathfinder-*.whl).FullName
+              (Get-Item "$env:CUDA_BINDINGS_ARTIFACTS_DIR/cuda_bindings-*.whl").FullName
+            )
           }
-          $pythonWheel = $pythonWheels[0].FullName
-          if ('${{ matrix.LOCAL_CTK }}' -eq '1') {
-            pip install --only-binary=:all: @dependencyArgs $pythonWheel
-          } else {
-            pip install --only-binary=:all: @dependencyArgs "${pythonWheel}[all]"
+          $pythonRequirements = @((Get-Item ./cuda_python*.whl).FullName)
+          if ('${{ matrix.LOCAL_CTK }}' -ne '1') {
+            $pythonRequirements = @($pythonRequirements | ForEach-Object { "$($_)[all]" })
           }
+          pip install --only-binary=:all: @dependencyArgs @pythonRequirements
 
       - name: Install cuda.pathfinder extra wheels for testing
         if: ${{ inputs.test-mode == 'standard' && inputs.test-pathfinder }}