test: batched jacobian now exists

avik-pal · avik-pal · commit e39ead45d45b · 2025-11-22T13:23:54.000-05:00
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -75,7 +75,6 @@ using Lux
             x -> x, AutoZygote(), rand(2), rand(2)
         )
 
-        @test_throws ArgumentError batched_jacobian(x -> x, AutoEnzyme(), rand(2, 2))
         @test_throws ErrorException batched_jacobian(x -> x, AutoZygote(), rand(2, 2))
     end
 

Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,6 @@ using Lux`
`75`	`75`	`x -> x, AutoZygote(), rand(2), rand(2)`
`76`	`76`	`)`
`77`	`77`
`78`		`- @test_throws ArgumentError batched_jacobian(x -> x, AutoEnzyme(), rand(2, 2))`
`79`	`78`	`@test_throws ErrorException batched_jacobian(x -> x, AutoZygote(), rand(2, 2))`
`80`	`79`	`end`
`81`	`80`