Skip to content

Commit e1a6fde

Browse files
committed
style fix
1 parent 3e249e4 commit e1a6fde

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

src/LinearMaps.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@ using SparseArrays
1010

1111
import Statistics: mean
1212

13-
import ChainRulesCore: rrule, frule, unthunk, @thunk, NoTangent
13+
using ChainRulesCore
14+
import ChainRulesCore: rrule
1415

1516
using Base: require_one_based_indexing
1617

src/chainrules.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,4 +16,4 @@ function rrule(A::LinearMap, x::AbstractVecOrMat)
1616
return NoTangent(), @thunk(A' * DY)
1717
end
1818
return y, pullback
19-
end
19+
end

test/rrules.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,4 +18,4 @@ using Flux, LinearAlgebra
1818
# Call rule
1919
g3 = gradient(x -> .5*norm(A'(x))^2, x)
2020
@test g1 g3[1]
21-
end
21+
end

0 commit comments

Comments
 (0)