Formatting

LuxDL · May 28, 2022 · 2d7ca99 · 2d7ca99
1 parent 681a27d
commit 2d7ca99
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/layers/basic.jl b/src/layers/basic.jl
@@ -727,15 +727,18 @@ function Base.show(io::IO, d::Scale)
     return print(io, ")")
 end
 
-function Scale(dims::Tuple{Vararg{Integer}}, activation=identity; init_weight=glorot_uniform,
+function Scale(dims::Tuple{Vararg{Integer}}, activation=identity;
+               init_weight=glorot_uniform,
                init_bias=zeros32, bias::Bool=true)
     activation = NNlib.fast_act(activation)
     return Scale{bias, typeof(activation), typeof(dims), typeof(init_weight),
                  typeof(init_bias)}(activation, dims, init_weight, init_bias)
 end
 
-Scale(s1::Integer, s23::Integer...; _act = identity, kw...) = Scale(tuple(s1, s23...), _act; kw...)
-Scale(size_act...; kw...) = Scale(size_act[1:end-1]...; _act = size_act[end], kw...)
+function Scale(s1::Integer, s23::Integer...; _act=identity, kw...)
+    Scale(tuple(s1, s23...), _act; kw...)
+end
+Scale(size_act...; kw...) = Scale(size_act[1:(end - 1)]...; _act=size_act[end], kw...)
 
 function initialparameters(rng::AbstractRNG, d::Scale{true})
     return (weight=d.init_weight(rng, d.dims...), bias=d.init_bias(rng, d.dims...))

diff --git a/test/layers/basic.jl b/test/layers/basic.jl
@@ -310,7 +310,8 @@ end
         end == [2.0 3.0; 4.0 5.0]
 
         @test begin
-            layer = Scale(2, tanh; bias = false, init_weight=zeros)
+            layer = Scale(2, tanh; bias=false, init_weight=zeros)
             first(Lux.apply(layer, [1 2; 3 4], Lux.setup(rng, layer)...))
         end == zeros(2, 2)
+    end
 end