From 48f507680ca6ab0afaf984b3735328a9a94c9d9c Mon Sep 17 00:00:00 2001
From: Charlie Marsh <charlie.r.marsh@gmail.com>
Date: Wed, 17 Sep 2025 10:18:30 -0400
Subject: [PATCH] Add PyG packages to torch backend (#15911)

## Summary

These are now supported on pyx.
---
 crates/uv-torch/src/backend.rs | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/crates/uv-torch/src/backend.rs b/crates/uv-torch/src/backend.rs
index b9fa72141..e3a5bd35f 100644
--- 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"