fix(tests): update spec dec conftest to use examples/dataset/make_dataset.py

ChenhanYu · ChenhanYu · commit 73dc213f1943 · 2026-04-07T10:39:35.000-07:00
prepare_input_conversations/ was moved to examples/dataset/ in this PR.
Update the test fixture to call ../dataset/make_dataset.py and use the
--full-conversations flag to match the README.

Signed-off-by: chenhany &lt;chenhany@nvidia.com&gt;
diff --git a/tests/examples/speculative_decoding/conftest.py b/tests/examples/speculative_decoding/conftest.py
@@ -36,7 +36,7 @@ def tiny_daring_anteater_path(tmp_path_factory):
     config_path.write_text(yaml.dump(config))
 
     run_example_command(
-        ["python", "prepare_input_conversations/make_dataset.py", "-f", str(config_path), "--full"],
+        ["python", "../dataset/make_dataset.py", "-f", str(config_path), "--full-conversations"],
         "speculative_decoding",
     )
 

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,7 @@ def tiny_daring_anteater_path(tmp_path_factory):`
`36`	`36`	`config_path.write_text(yaml.dump(config))`
`37`	`37`
`38`	`38`	`run_example_command(`
`39`		`- ["python", "prepare_input_conversations/make_dataset.py", "-f", str(config_path), "--full"],`
	`39`	`+ ["python", "../dataset/make_dataset.py", "-f", str(config_path), "--full-conversations"],`
`40`	`40`	`"speculative_decoding",`
`41`	`41`	`)`
`42`	`42`