From c598f86476fb84f6f9da75611667347f0bf08c1b Mon Sep 17 00:00:00 2001 From: Charlie Marsh Date: Tue, 14 May 2024 20:02:09 -0400 Subject: [PATCH] Allow local versions in wheel filenames (#3596) ## Summary Closes https://github.com/astral-sh/uv/issues/3594. ## Test Plan `cargo run pip install --verbose https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl --no-deps` --- crates/install-wheel-rs/src/linker.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/crates/install-wheel-rs/src/linker.rs b/crates/install-wheel-rs/src/linker.rs index 8c0a2be8f..a71b8a32d 100644 --- a/crates/install-wheel-rs/src/linker.rs +++ b/crates/install-wheel-rs/src/linker.rs @@ -52,7 +52,7 @@ pub fn install_wheel( } let version = Version::from_str(&version)?; - if version != filename.version { + if version != filename.version && version != filename.version.clone().without_local() { return Err(Error::MismatchedVersion(version, filename.version.clone())); } }