need long

Jackmin801 · Jackmin801 · commit ad919595b1af · 2025-06-10T23:27:45.000-07:00
diff --git a/src/zeroband/utils/parquet.py b/src/zeroband/utils/parquet.py
@@ -17,6 +17,6 @@
         ("step", pa.int32()),
         ("target_lengths", pa.int32()),
         ("task_type", pa.string()),
-        ("seed", pa.int32()),  # Optional - can be null
+        ("seed", pa.int64()),  # Optional - can be null
     ]
 )

Original file line number	Diff line number	Diff line change
`@@ -17,6 +17,6 @@`
`17`	`17`	`("step", pa.int32()),`
`18`	`18`	`("target_lengths", pa.int32()),`
`19`	`19`	`("task_type", pa.string()),`
`20`		`- ("seed", pa.int32()), # Optional - can be null`
	`20`	`+ ("seed", pa.int64()), # Optional - can be null`
`21`	`21`	`]`
`22`	`22`	`)`