chore: address more CR

Signed-off-by: mkornfield <mkornfield@nvidia.com>

Author: mckornfield

.github/dco.yaml

@@ -159,7 +159,7 @@
     "    .with_data_source(df)  # .with_replace_pii(enable=False) to disable PII replacement\n",
     "    .with_differential_privacy(dp_enabled=True, delta=recommended_delta)\n",
     "    .with_train(batch_size=16)  # Override the default batch size of 1, which is designed for non-DP training\n",
-    "    .with_generate(structured_generation={"enabled": True})  # Improves the percentage of valid records when DP is enabled\n",
+    "    .with_generate(structured_generation={'enabled': True})  # Improves the percentage of valid records when DP is enabled\n",
     ")\n",
     "builder.run()\n",
     "results = builder.results"

docs/tutorials/differential-privacy.ipynb

@@ -379,7 +379,7 @@ Generation stopped prematurely because the average fraction of invalid records w
 ```
 
 : Too many invalid records across `generation.patience` consecutive batches.
-  Consider retraining with more records, adjusting `training.number_of_input_records_to_sample`, or setting `generation.structured_generation.enabled=true`.
+  Consider retraining with more records, adjusting `training.num_input_records_to_sample`, or setting `generation.structured_generation.enabled=true`.
 
 For context-length errors during data assembly (`"The number of tokens in an
 example exceeds the available context length"`), see

docs/user-guide/troubleshooting.md

@@ -494,7 +494,31 @@ def cmd(**kwargs):
     assert captured["generation"]["structured_generation"]["backend"] == "outlines"
 
 
-def test_structured_generation_legacy_option_end_to_end_via_click_runner():
+@pytest.mark.parametrize(
+    ("option", "legacy_key", "value", "expected"),
+    [
+        ("--generation__use_structured_generation", "use_structured_generation", "true", True),
+        ("--generation__structured_generation_backend", "structured_generation_backend", "outlines", "outlines"),
+        (
+            "--generation__structured_generation_schema_method",
+            "structured_generation_schema_method",
+            "json_schema",
+            "json_schema",
+        ),
+        (
+            "--generation__structured_generation_use_single_sequence",
+            "structured_generation_use_single_sequence",
+            "true",
+            True,
+        ),
+    ],
+)
+def test_structured_generation_legacy_options_end_to_end_via_click_runner(
+    option: str,
+    legacy_key: str,
+    value: str,
+    expected: object,
+):
     """Legacy flat structured-generation CLI aliases remain accepted during migration."""
     captured: dict = {}
 
@@ -503,6 +527,6 @@ def test_structured_generation_legacy_option_end_to_end_via_click_runner():
     def cmd(**kwargs):
         captured.update(parse_overrides(kwargs))
 
-    result = CliRunner().invoke(cmd, ["--generation__structured_generation_backend", "outlines"])
+    result = CliRunner().invoke(cmd, [option, value])
     assert result.exit_code == 0, result.output
-    assert captured["generation"]["structured_generation_backend"] == "outlines"
+    assert captured["generation"][legacy_key] == expected