Update EvaluationRow model to set default for pid and simplify TestFunction type definition

Dylan Huang · Dylan Huang · commit 873a78a00cbf · 2025-08-21T21:39:28.000-07:00
diff --git a/eval_protocol/models.py b/eval_protocol/models.py
@@ -541,7 +541,7 @@ class EvaluationRow(BaseModel):
     )
 
     pid: Optional[int] = Field(
-        None,
+        default=None,
         description="The PID of the process that created the row. This is used by the evaluation watcher to detect stopped evaluations.",
     )
 
diff --git a/eval_protocol/pytest/types.py b/eval_protocol/pytest/types.py
@@ -31,7 +31,7 @@
 """
 Test function types
 """
-TestFunction = Callable[..., Dataset]
+TestFunction = Callable
 
 """
 Rollout processor types

Original file line number	Diff line number	Diff line change
`@@ -541,7 +541,7 @@ class EvaluationRow(BaseModel):`
`541`	`541`	`)`
`542`	`542`
`543`	`543`	`pid: Optional[int] = Field(`
`544`		`- None,`
	`544`	`+ default=None,`
`545`	`545`	`description="The PID of the process that created the row. This is used by the evaluation watcher to detect stopped evaluations.",`
`546`	`546`	`)`
`547`	`547`