Skip Jax dependent test if on Windows (#661)

(conv_2d_no_bias)
microsoft · Mar 19, 2021 · 0bdedff · 0bdedff
1 parent 31b47b9
commit 0bdedff
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/python/test_tracing_nn.py b/test/python/test_tracing_nn.py
@@ -4,8 +4,10 @@
 import ksc
 from ksc.tracing.functions import math, nn
 from ksc.shape import Shape, TensorShape, ScalarShape
+import sys
 
-
+# https://github.com/google/jax/issues/5795
+@pytest.mark.skipif(sys.platform == 'win32', reason="Jaxlib builds not currently supplied for Windows")
 @pytest.mark.parametrize(
     "kernel_size,strides,padding,expected_shape", [
         ((3, 3), (1, 1), "SAME", (1, 64, 56, 56)),