Skip to content

Commit 510c4b2

Browse files
ChenhanYuclaude
ChenhanYu
and
claude
committed
refactor(examples): move dataset prep scripts to examples/dataset/
Move all dataset preparation scripts out of the speculative-decoding- specific prepare_input_conversations/ folder into a top-level examples/dataset/ directory to signal that these utilities are reusable beyond speculative decoding. - Rename _specdec_aug.py → conversation_utils.py (general-purpose name) - Update imports in make_nemotron_ptv{2,3}_dataset.py - Update speculative_decoding/README.md paths to ../dataset/ Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> Signed-off-by: chenhany <chenhany@nvidia.com>
1 parent 175b334 commit 510c4b2

10 files changed

examples/speculative_decoding/prepare_input_conversations/__init__.py renamed to examples/dataset/__init__.py

File renamed without changes.

examples/speculative_decoding/prepare_input_conversations/add_nemotron_chat.py renamed to examples/dataset/add_nemotron_chat.py

File renamed without changes.

examples/speculative_decoding/prepare_input_conversations/augmentations.yaml renamed to examples/dataset/augmentations.yaml

File renamed without changes.

examples/speculative_decoding/prepare_input_conversations/_specdec_aug.py renamed to examples/dataset/conversation_utils.py

File renamed without changes.

examples/speculative_decoding/prepare_input_conversations/example_data_config.yaml renamed to examples/dataset/example_data_config.yaml

File renamed without changes.

examples/speculative_decoding/prepare_input_conversations/make_dataset.py renamed to examples/dataset/make_dataset.py

File renamed without changes.

examples/speculative_decoding/prepare_input_conversations/make_nemotron_ptv2_dataset.py renamed to examples/dataset/make_nemotron_ptv2_dataset.py

File renamed without changes.

examples/speculative_decoding/prepare_input_conversations/make_nemotron_ptv3_dataset.py renamed to examples/dataset/make_nemotron_ptv3_dataset.py

File renamed without changes.

examples/speculative_decoding/prepare_input_conversations/nemotron_ptv3_datasets.yaml renamed to examples/dataset/nemotron_ptv3_datasets.yaml

File renamed without changes.

examples/speculative_decoding/prepare_input_conversations/utils.py renamed to examples/dataset/utils.py

File renamed without changes.

0 commit comments

Comments
 (0)