formatter

SciML · Dec 28, 2024 · 3eacc11 · 3eacc11
1 parent 6b24bab
commit 3eacc11
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/models/nomad.jl b/src/models/nomad.jl
@@ -41,9 +41,9 @@ julia> size(first(nomad((u, y), ps, st)))
 ```
 """
 @concrete struct NOMAD <: AbstractLuxContainerLayer{(:approximator, :decoder)}
-  approximator
-  decoder
-  concatenate <: Function
+    approximator
+    decoder
+    concatenate <: Function
 end
 
 """
@@ -104,7 +104,9 @@ function (nomad::NOMAD)(x, ps, st::NamedTuple)
     return out, (approximator=st_a, decoder=st_d)
 end
 
-NOMAD(approximator_net, decoder_net; concatenate=nomad_concatenate) = NOMAD(approximator_net, decoder_net, concatenate)
+function NOMAD(approximator_net, decoder_net; concatenate=nomad_concatenate)
+    NOMAD(approximator_net, decoder_net, concatenate)
+end
 
 batch_vectorize(x::AbstractArray) = reshape(x, :, size(x, ndims(x)))