fix inadvertent corruption of src

FluxML · Jun 11, 2024 · 16d8c17 · 16d8c17
1 parent f658765
commit 16d8c17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/regressor.jl b/src/regressor.jl
@@ -49,7 +49,7 @@ shape(model::MultitargetNeuralNetworkRegressor, X, y) = (ncols(X), ncols(y))
 build(model::MultitargetNeuralNetworkRegressor, rng, shape) =
     build(model.builder, rng, shape...)
 
-function fitresult(model::MultitargetNeuralNetworkRegressor, chaintype, y)
+function fitresult(model::MultitargetNeuralNetworkRegressor, chain, y)
     if y isa Matrix
         target_column_names = nothing
     else