Skip to content

feat: add generic Hugging Face dataset converter - #9653

Draft
joseph-isaacs wants to merge 3 commits into
developfrom
ji/hf-dataset-vortex-converter
Draft

feat: add generic Hugging Face dataset converter#9653
joseph-isaacs wants to merge 3 commits into
developfrom
ji/hf-dataset-vortex-converter

Conversation

@joseph-isaacs

@joseph-isaacs joseph-isaacs commented Aug 27, 2026

Copy link
Copy Markdown
Contributor

Summary

  • add a resumable converter for Parquet shards in any Hugging Face dataset repository
  • support Parquet Zstd, Vortex, and compact Vortex outputs with deterministic destination paths
  • require Xet-backed source and destination repositories and expose bounded Xet download concurrency
  • bound prefetched source data and pending upload data with independent worker pools
  • batch Hub commits, verify downloads, checkpoint progress, and delete local outputs only after successful commits
  • add a Linux process-tree resource monitor for CPU, memory, disk, and network usage
  • redirect monitored child output to a durable log and make progress logging safe after terminal detachment
  • distinguish transfer failures from logging failures and reuse local encodings during checkpoint resume
  • preserve checkpoint compatibility when operational Xet cache and concurrency settings change

Validation

  • data/fineweb-convert/.venv/bin/python -m unittest scripts/tests/test_hf_parquet_convert.py (17 tests)
  • python3.12 -m py_compile scripts/huggingface-convert.py scripts/monitor-process.py scripts/tests/test_hf_parquet_convert.py
  • closed-stdout end-to-end monitor test with durable child-log verification
  • live checkpoint resume: skipped 120 committed shards and reused pending local encodings
  • converter CLI help inspection
  • resource-monitor smoke test
  • git diff --check

The converter has also been exercised against a large FineWeb upload, but it contains no FineWeb-specific repository or path assumptions.

Signed-off-by: Joe Isaacs <joe.isaacs@live.co.uk>
@codspeed-hq

codspeed-hq Bot commented Aug 27, 2026

Copy link
Copy Markdown

Merging this PR will regress 1 benchmark

⚠️ Unknown Walltime execution environment detected

Using the Walltime instrument on standard Hosted Runners will lead to inconsistent data.

For the most accurate results, we recommend using CodSpeed Macro Runners: bare-metal machines fine-tuned for performance measurement consistency.

⚠️ Different runtime environments detected

Some benchmarks with significant performance changes were compared across different runtime environments,
which may affect the accuracy of the results.

Open the report in CodSpeed to investigate

⚡ 3 improved benchmarks
❌ 1 regressed benchmark
✅ 2126 untouched benchmarks
⏩ 106 skipped benchmarks1
🗄️ 4 archived benchmarks run2

Warning

Please fix the performance issues or acknowledge them on CodSpeed.

Performance Changes

Mode Benchmark BASE HEAD Efficiency
WallTime mul_u32_nonnull_avx512 5.6 µs 6.3 µs -11.87%
WallTime arrow_checked_add_u32_avx2[16384] 21.3 µs 17.7 µs +20.53%
Simulation take[duplicates/repeated/primitive/nonnull/chunks=16/indices=1000] 245.7 µs 205.4 µs +19.61%
WallTime arrow_checked_add_u32_neon[16384] 13.5 µs 12.2 µs +10.52%

Tip

Investigate this regression by commenting @codspeedbot fix this regression on this PR, or directly use the CodSpeed MCP with your agent.


Comparing ji/hf-dataset-vortex-converter (59f79b4) with develop (59a056d)

Open in CodSpeed

Footnotes

  1. 106 benchmarks were skipped, so the baseline results were used instead. If they were deleted from the codebase, click here and archive them to remove them from the performance reports.

  2. 4 benchmarks were run, but are now archived. If they were deleted in another branch, consider rebasing to remove them from the report. Instead if they were added back, click here to restore them.

Signed-off-by: Joe Isaacs <joe.isaacs@live.co.uk>
Signed-off-by: Joe Isaacs <joe.isaacs@live.co.uk>
joseph-isaacs added a commit to vortex-data/datasets that referenced this pull request Aug 27, 2026
Cleaned-up port of scripts/huggingface-convert.py from vortex-data/vortex#9653.
The script now takes a list of external source repositories (--sources JSON or
a single --repo), diffs each selected Parquet shard against the same-named path
in the destination vortex-data repository, and downloads/converts/uploads only
the missing files. Adds --dry-run diff reporting, per-source checkpoints and
metrics, a sources.example.json, tests, and a README.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01GzCzGvYf8WQcyUnQLc7cZS
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant