Update src/mlj_model_interface.jl

Co-authored-by: Okon Samuel <39421418+OkonSamuel@users.noreply.github.com>
FluxML · Sep 4, 2023 · 98de344 · 98de344
1 parent bbc1a44
commit 98de344
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mlj_model_interface.jl b/src/mlj_model_interface.jl
@@ -59,7 +59,7 @@ function MLJModelInterface.fit(model::MLJFluxModel,
         build(model, rng, shape) |> move
     catch ex
         @error ERR_BUILDER
-        throw(ex)
+        rethrow()
     end
 
     penalty = Penalty(model)