Run nn_image_classification_winml only when winml is enabled.

bytecodealliance · Feb 5, 2024 · 4b456ec · 4b456ec
1 parent 9732234
commit 4b456ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crates/wasi-nn/tests/all.rs b/crates/wasi-nn/tests/all.rs
@@ -95,7 +95,7 @@ fn nn_image_classification_named() {
     run(NN_IMAGE_CLASSIFICATION_NAMED, true).unwrap()
 }
 
-#[cfg_attr(not(all(target_arch = "x86_64", target_os = "windows")), ignore)]
+#[cfg_attr(not(feature = "winml"), ignore)]
 #[test]
 fn nn_image_classification_winml() {
     run(NN_IMAGE_CLASSIFICATION_WINML, true).unwrap()