Add PyG packages to torch backend (#15911)

## Summary These are now supported on pyx.
2025-09-17 10:18:30 -04:00 · 2025-09-17 10:18:30 -04:00 · 48f507680c
parent d4806ee921
commit 48f507680c
1 changed files with 7 additions and 2 deletions
--- a/crates/uv-torch/src/backend.rs
+++ b/crates/uv-torch/src/backend.rs
@ -322,11 +322,16 @@ impl TorchStrategy {
            TorchSource::Pyx => {
                matches!(
                    package_name.as_str(),
-                    "flash-attn"
+                    "deepspeed"
+                        | "flash-attn"
                        | "flash-attn-3"
                        | "megablocks"
                        | "natten"
-                        | "deepspeed"
+                        | "pyg-lib"
+                        | "torch-cluster"
+                        | "torch-scatter"
+                        | "torch-sparse"
+                        | "torch-spline-conv"
                        | "vllm"
                        | "torch"
                        | "torch-model-archiver"