add seed to conftest

Jackmin801 · Jackmin801 · commit 216f51a65e0b · 2025-06-11T18:25:48.000-07:00
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -107,6 +107,7 @@ def create_dummy_parquet_table(batch_size: int, seq_len: int) -> Table:
         "problem_id": pa.array(["0"] * batch_size, type=pa.string()),
         "input_logprobs": pa.array([[0.1] * seq_len for _ in range(batch_size)], type=pa.list_(pa.float32())),
         "output_logprobs": pa.array([[0.1] * seq_len for _ in range(batch_size)], type=pa.list_(pa.float32())),
+        "seed": pa.array([42] * batch_size, type=pa.int64()),
     }
 
     # Create table directly from dictionary

Original file line number	Diff line number	Diff line change
`@@ -107,6 +107,7 @@ def create_dummy_parquet_table(batch_size: int, seq_len: int) -> Table:`
`107`	`107`	`"problem_id": pa.array(["0"] * batch_size, type=pa.string()),`
`108`	`108`	`"input_logprobs": pa.array([[0.1] * seq_len for _ in range(batch_size)], type=pa.list_(pa.float32())),`
`109`	`109`	`"output_logprobs": pa.array([[0.1] * seq_len for _ in range(batch_size)], type=pa.list_(pa.float32())),`
	`110`	`+ "seed": pa.array([42] * batch_size, type=pa.int64()),`
`110`	`111`	`}`
`111`	`112`
`112`	`113`	`# Create table directly from dictionary`