Fix AI slop in migration guide "Inference After Training"

lukstafi · lukstafi · commit 4b4f6d7eb641 · 2025-10-18T21:31:06.000+08:00
diff --git a/docs/migration_guide.md b/docs/migration_guide.md
@@ -429,20 +429,23 @@ def inference(model, test_input):
 
 **OCANNL:**
 ```ocaml
-let inference ctx model test_input =
+let inference ctx model =
   (* Define inference computation - use %cd to avoid initialization *)
   let%cd output = model { test_input } in
   Train.set_on_host output.value;
   
   (* Compile inference routine *)
-  let infer_routine = Train.to_routine ctx IDX.empty output in
+  let ctx, routine = Context.compile ctx output IDX.empty in
   
-  (* Run inference *)
-  Tn.set_values test_input.value input_data;
-  Train.run ctx infer_routine;
+  fun input_data ->
+    (* Run inference *)
+    Tn.set_values test_input.value input_data;
+    Train.run ctx infer_routine;
   
-  (* Extract predictions *)
-  output.@{[| (* indices *) |]}
+    (* Before OCANNL v0.7, to get all cells flattened: *)
+    Tn.get_values output.value
+    (* Or starting with the future OCANNL v0.7 to get a bigarray: *)
+    Context.get ctx output.value
 ```
 
 ### Key API Functions
