diff --git a/crates/uv-resolver/src/resolver/mod.rs b/crates/uv-resolver/src/resolver/mod.rs index d0f43e9c9..92f9fba38 100644 --- a/crates/uv-resolver/src/resolver/mod.rs +++ b/crates/uv-resolver/src/resolver/mod.rs @@ -1839,6 +1839,7 @@ impl ResolverState bool { + matches!( + package_name.as_str(), + "torch" + | "torcharrow" + | "torchaudio" + | "torchcsprng" + | "torchdata" + | "torchdistx" + | "torchtext" + | "torchvision" + ) + } + /// Return the appropriate index URLs for the given [`TorchStrategy`]. pub fn index_urls(&self) -> impl Iterator { match self {