Unverified Commit d3ee09d7 authored by Alec's avatar Alec Committed by GitHub
Browse files

fix: gracefully skip NIXL UCX tests when backend unavailable (#6484)


Signed-off-by: default avataralec-flowers <aflowers@nvidia.com>
Co-authored-by: default avatarClaude Opus 4.6 <noreply@anthropic.com>
parent a9568455
...@@ -235,7 +235,13 @@ mod tests { ...@@ -235,7 +235,13 @@ mod tests {
#[test] #[test]
fn test_require_backend() { fn test_require_backend() {
let agent = NixlAgent::new_with_backends("test", &["UCX"]).expect("Need UCX for test"); let agent = match NixlAgent::new_with_backends("test", &["UCX"]) {
Ok(agent) => agent,
Err(_) => {
eprintln!("Skipping test_require_backend: UCX not available");
return;
}
};
// Should succeed for available backend // Should succeed for available backend
assert!(agent.require_backend("UCX").is_ok()); assert!(agent.require_backend("UCX").is_ok());
...@@ -247,8 +253,13 @@ mod tests { ...@@ -247,8 +253,13 @@ mod tests {
#[test] #[test]
fn test_require_backends_strict() { fn test_require_backends_strict() {
// Should succeed if UCX is available // Should succeed if UCX is available
let agent = NixlAgent::require_backends("test_strict", &["UCX"]) let agent = match NixlAgent::require_backends("test_strict", &["UCX"]) {
.expect("Failed to require backends"); Ok(agent) => agent,
Err(_) => {
eprintln!("Skipping test_require_backends_strict: UCX not available");
return;
}
};
assert!(agent.has_backend("UCX")); assert!(agent.has_backend("UCX"));
// Should fail if any backend is missing (GDS likely not available) // Should fail if any backend is missing (GDS likely not available)
......
...@@ -184,7 +184,16 @@ mod tests { ...@@ -184,7 +184,16 @@ mod tests {
..Default::default() ..Default::default()
}; };
let loader: MediaLoader = MediaLoader::new(media_decoder, Some(fetcher)).unwrap(); let loader: MediaLoader = match MediaLoader::new(media_decoder, Some(fetcher)) {
Ok(l) => l,
Err(e) => {
println!(
"test test_fetch_and_decode ... ignored (NIXL/UCX not available: {})",
e
);
return;
}
};
let image_url = ImageUrl::from(format!("{}/llm-optimize-deploy-graphic.png", server.url())); let image_url = ImageUrl::from(format!("{}/llm-optimize-deploy-graphic.png", server.url()));
let content_part = ChatCompletionRequestUserMessageContentPart::ImageUrl( let content_part = ChatCompletionRequestUserMessageContentPart::ImageUrl(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment