Skip to content
This repository was archived by the owner on Sep 28, 2024. It is now read-only.

Commit b79fd32

Browse files
committed
fix style
1 parent b89b464 commit b79fd32

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

src/fourier.jl

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ end
6868

6969
Flux.@functor SpectralConv
7070

71-
Base.ndims(::SpectralConv{P,N}) where {P,N} = N
71+
Base.ndims(::SpectralConv{P, N}) where {P, N} = N
7272

7373
function Base.show(io::IO, l::SpectralConv{P}) where {P}
7474
print(io, "SpectralConv($(l.in_channel) => $(l.out_channel), $(l.modes), σ=$(string(l.σ)), permuted=$P)")
@@ -150,7 +150,15 @@ end
150150
Flux.@functor FourierOperator
151151

152152
function Base.show(io::IO, l::FourierOperator)
153-
print(io, "FourierOperator($(l.conv.in_channel) => $(l.conv.out_channel), $(l.conv.modes), σ=$(string(l.σ)), permuted=$(l.conv.permuted))")
153+
print(
154+
io,
155+
"FourierOperator(" *
156+
"$(l.conv.in_channel) => $(l.conv.out_channel), " *
157+
"$(l.conv.modes), " *
158+
"σ=$(string(l.σ)), " *
159+
"permuted=$(l.conv.permuted)" *
160+
")"
161+
)
154162
end
155163

156164
function (m::FourierOperator)(𝐱)

0 commit comments

Comments
 (0)