Mark some of the tests as broken

LuxDL · Jun 13, 2024 · d485aa0 · d485aa0
1 parent ebdc381
commit d485aa0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/Project.toml b/Project.toml
@@ -43,6 +43,7 @@ Artifacts = "1.10"
 ChainRulesCore = "1.24"
 ComponentArrays = "0.15.13"
 ConcreteStructs = "0.2.3"
+DataInterpolations = "5.2.0"
 ExplicitImports = "1.5"
 ForwardDiff = "0.10.36"
 GPUArraysCore = "0.1.6"

diff --git a/test/layer_tests.jl b/test/layer_tests.jl
@@ -150,8 +150,8 @@ end
         ongpu && continue
 
         @testset "$(spl): train_grid $(train_grid), dims $(dims)" for spl in (
-                ConstantInterpolation, LinearInterpolation, QuadraticInterpolation,
-                QuadraticSpline, CubicSpline),
+                ConstantInterpolation, LinearInterpolation,
+                QuadraticInterpolation, QuadraticSpline, CubicSpline),
             train_grid in (true, false),
             dims in ((), (8,))
 
@@ -164,12 +164,14 @@ end
             y, st = spline(x, ps, st)
             @test size(y) == (dims..., 4)
 
-            @jet spline(x, ps, st)
+            opt_broken = !ongpu && dims != () && spl !== ConstantInterpolation
+
+            @jet spline(x, ps, st) opt_broken=opt_broken # See SciML/DataInterpolations.jl/issues/267
 
             y, st = spline(x, ps_ca, st)
             @test size(y) == (dims..., 4)
 
-            @jet spline(x, ps_ca, st)
+            @jet spline(x, ps_ca, st) opt_broken=opt_broken # See SciML/DataInterpolations.jl/issues/267
 
             ∂x, ∂ps = Zygote.gradient((x, ps) -> sum(abs2, first(spline(x, ps, st))), x, ps)
             spl !== ConstantInterpolation && @test ∂x !== nothing