Make getindex on MetricProblem return a MetricProblem

Herb-AI · Oct 18, 2024 · cac6251 · cac6251
1 parent dda32d1
commit cac6251
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/problem.jl b/src/problem.jl
@@ -99,6 +99,6 @@ end
 Overwrite `Base.getindex` to allow for slicing of input/output-based problems.
 """
 Base.getindex(p::Problem{Vector{IOExample}}, indices) = Problem(p.spec[indices])
-Base.getindex(p::MetricProblem{Vector{IOExample}}, indices) = Problem(p.spec[indices])
+Base.getindex(p::MetricProblem{Vector{IOExample}}, indices) = MetricProblem(p.cost_function, p.spec[indices])