update

Author: arjkesh

.github/workflows/build-wheel.yml

@@ -26,6 +26,11 @@ on:
         required: false
         type: number
         default: 7
+      skip-import-test:
+        description: 'Skip import test (for CUDA/GPU-dependent packages)'
+        required: false
+        type: boolean
+        default: false
     outputs:
       artifact-name:
         description: 'Name of the uploaded wheel artifact'
@@ -61,6 +66,7 @@ jobs:
           twine check dist/*
 
       - name: Test import
+        if: ${{ !inputs.skip-import-test }}
         run: |
           pip install dist/*.whl
           python -c "import ${{ inputs.package-name }}; print('✅ Imported successfully')"

.github/workflows/tilegym-ci.yml

@@ -158,13 +158,15 @@ jobs:
   build-wheel:
     name: build-python-wheel
     needs: config
+    if: needs.config.outputs.build == 'true'
     uses: ./.github/workflows/build-wheel.yml
     with:
       package-name: tilegym
       # Artifact naming: PR builds -> tilegym-pr-wheel-{sha}, Main -> tilegym-wheel-{sha}
       artifact-suffix: ${{ needs.config.outputs.is_pr == 'true' && '-pr' || '' }}
       python-version: '3.10'
       retention-days: 7
+      skip-import-test: true  # TileGym requires CUDA, test in Docker instead
 
   build:
     name: build-tilegym-image