Skip to content

Commit

Permalink
format fix
Browse files Browse the repository at this point in the history
  • Loading branch information
BioTurboNick committed Sep 16, 2024
1 parent 250fec9 commit 15a5050
Showing 1 changed file with 10 additions and 6 deletions.
16 changes: 10 additions & 6 deletions test/rulesets/Base/indexing.jl
Original file line number Diff line number Diff line change
Expand Up @@ -217,16 +217,20 @@ 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); check_inferred=false)
test_rrule(collecteachrow, rand(3, 4); check_inferred=false)
test_rrule(collect eachrow, rand(5); check_inferred=false)
test_rrule(collect eachrow, rand(3, 4); check_inferred=false)

test_rrule(collecteachcol, rand(3, 4); check_inferred=false)
@test_skip test_rrule(collecteachcol, Diagonal(rand(5))) # works locally!
test_rrule(collect eachcol, rand(3, 4); check_inferred=false)
@test_skip test_rrule(collect eachcol, 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); check_inferred=false, fkwargs = (; dims = 3))
test_rrule(collecteachslice, rand(3, 4, 5); check_inferred=false, fkwargs = (; dims = (2,)))
test_rrule(
collect eachslice, rand(3, 4, 5); check_inferred=false, fkwargs = (; dims = 3)
)
test_rrule(
collect eachslice, rand(3, 4, 5); check_inferred=false, fkwargs = (; dims = (2,))
)

test_rrule(
collect eachslice,
Expand Down

0 comments on commit 15a5050

Please sign in to comment.