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",