Skip to content

Commit

Permalink
turn off inference check
Browse files Browse the repository at this point in the history
  • Loading branch information
BioTurboNick committed Sep 16, 2024
1 parent 0c0ce2b commit 250fec9
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions test/rulesets/Base/indexing.jl
Original file line number Diff line number Diff line change
Expand Up @@ -217,16 +217,16 @@ end
# DimensionMismatch("second dimension of A, 6, does not match length of x, 5")
# Probably similar to https://github.com/JuliaDiff/ChainRulesTestUtils.jl/issues/234 (about Broadcasted not Generator)

test_rrule(collecteachrow, rand(5))
test_rrule(collecteachrow, rand(3, 4))
test_rrule(collecteachrow, rand(5); check_inferred=false)
test_rrule(collecteachrow, rand(3, 4); check_inferred=false)

test_rrule(collecteachcol, rand(3, 4))
test_rrule(collecteachcol, rand(3, 4); check_inferred=false)
@test_skip test_rrule(collecteachcol, Diagonal(rand(5))) # works locally!

if VERSION >= v"1.7"
# On 1.6, ComposedFunction doesn't take keywords. Only affects this testing strategy, not real use.
test_rrule(collecteachslice, rand(3, 4, 5); fkwargs = (; dims = 3))
test_rrule(collecteachslice, rand(3, 4, 5); fkwargs = (; dims = (2,)))
test_rrule(collecteachslice, rand(3, 4, 5); check_inferred=false, fkwargs = (; dims = 3))
test_rrule(collecteachslice, rand(3, 4, 5); check_inferred=false, fkwargs = (; dims = (2,)))

test_rrule(
collect eachslice,
Expand Down

0 comments on commit 250fec9

Please sign in to comment.