Fix doctrine dataset load (explicit jsonl)

train_willay.py

@@ -74,7 +74,11 @@ def to_msgs(u, a):
 
 
 def main():
-    doctrine = load_dataset("SZLHOLDINGS/szl-1-doctrine-sft", split="train")
+    doctrine = load_dataset(
+        "json",
+        data_files="hf://datasets/SZLHOLDINGS/szl-1-doctrine-sft/szl_dataset.jsonl",
+        split="train",
+    )
     refusal = Dataset.from_list([to_msgs(u, a) for u, a in REFUSALS])
     # upsample the refusal set so the specialist behavior dominates
     combined = concatenate_datasets([doctrine, refusal, refusal, refusal])