Update

lanl-ansi · Jun 6, 2024 · 224f5b3 · 224f5b3
1 parent e179d52
commit 224f5b3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/models/Pipeline.jl b/src/models/Pipeline.jl
@@ -20,10 +20,10 @@ julia> model = Model();
 
 julia> @variable(model, x[1:2]);
 
-julia> f = Omelette.Pipeline([
+julia> f = Omelette.Pipeline(
            Omelette.LinearRegression([1.0 2.0], [0.0]),
            Omelette.ReLUQuadratic(),
-       ])
+       )
 Omelette.Pipeline(Omelette.AbstractPredictor[Omelette.LinearRegression([1.0 2.0], [0.0]), Omelette.ReLUQuadratic()])
 
 julia> y = Omelette.add_predictor(model, f, x)
@@ -44,6 +44,8 @@ struct Pipeline <: AbstractPredictor
     layers::Vector{AbstractPredictor}
 end
 
+Pipeline(args::AbstractPredictor...) = Pipeline(collect(args))
+
 function add_predictor(
     model::JuMP.Model,
     predictor::Pipeline,