diff --git a/examples/src/main/java/dev/braintrust/examples/ExperimentExample.java b/examples/src/main/java/dev/braintrust/examples/ExperimentExample.java index cfd9a1e8..41eaf9d6 100644 --- a/examples/src/main/java/dev/braintrust/examples/ExperimentExample.java +++ b/examples/src/main/java/dev/braintrust/examples/ExperimentExample.java @@ -41,7 +41,7 @@ public static void main(String[] args) throws Exception { EvalCase.of("asparagus", "vegetable"), EvalCase.of("apple", "fruit"), EvalCase.of("banana", "fruit")) - .task(getFoodType) + .taskFunction(getFoodType) .scorers( Scorer.of( "fruit_scorer", diff --git a/src/main/java/dev/braintrust/eval/Eval.java b/src/main/java/dev/braintrust/eval/Eval.java index 11b92a5f..01d089a2 100644 --- a/src/main/java/dev/braintrust/eval/Eval.java +++ b/src/main/java/dev/braintrust/eval/Eval.java @@ -250,8 +250,8 @@ public Builder task(Task task) { return this; } - public Builder task(Function taskFn) { - return task((Task) evalCase -> taskFn.apply(evalCase.input())); + public Builder taskFunction(Function taskFn) { + return task(evalCase -> taskFn.apply(evalCase.input())); } @SafeVarargs diff --git a/src/test/java/dev/braintrust/eval/EvalTest.java b/src/test/java/dev/braintrust/eval/EvalTest.java index 873521de..ca974162 100644 --- a/src/test/java/dev/braintrust/eval/EvalTest.java +++ b/src/test/java/dev/braintrust/eval/EvalTest.java @@ -6,7 +6,6 @@ import dev.braintrust.api.BraintrustApiClient; import dev.braintrust.trace.BraintrustTracing; import io.opentelemetry.api.common.AttributeKey; -import java.util.function.Function; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; @@ -30,7 +29,7 @@ public void evalOtelTraceWithProperAttributes() { .cases( EvalCase.of("strawberry", "fruit"), EvalCase.of("asparagus", "vegetable")) - .task((Function) food -> "fruit") + .task(food -> "fruit") .scorers( Scorer.of( "fruit_scorer",