Skip to content

Update AG News dataset source to fancyzhx/ag_news - #4

Merged
nevil-mathew merged 2 commits into
claude/repo-assessment-completeness-ipnli2from
claude/topic-extraction-benchmark-error-juhs58
Jul 4, 2026
Merged

Update AG News dataset source to fancyzhx/ag_news#4
nevil-mathew merged 2 commits into
claude/repo-assessment-completeness-ipnli2from
claude/topic-extraction-benchmark-error-juhs58

Conversation

@nevil-mathew

Copy link
Copy Markdown
Owner

Summary

Updates the AG News dataset source from the default ag_news to the fancyzhx/ag_news dataset on Hugging Face Hub across benchmark and notebook files.

Changes

  • Updated run_benchmark.py to load AG News from fancyzhx/ag_news instead of ag_news
    • Modified documentation comment to reflect new dataset identifier
    • Updated load_ag_news() function to use the new dataset path
  • Updated cumulative_tritopic_benchmark.ipynb notebook to use fancyzhx/ag_news dataset identifier

Details

This change ensures consistent dataset sourcing across the benchmark suite by explicitly specifying the dataset namespace. The fancyzhx/ag_news dataset is the maintained version on Hugging Face Hub and provides better reliability and version control compared to the default ag_news identifier.

https://claude.ai/code/session_01LCniwV83zZFgkbEYU5Jwcf

HF's datasets/huggingface_hub stopped resolving the legacy script-based
"ag_news" repo id (HfUriError: repo id must be namespace/name), breaking
the full benchmark CI job. Point at fancyzhx/ag_news, the parquet-based
mirror with the same text/label schema.
@coderabbitai

coderabbitai Bot commented Jul 4, 2026

Copy link
Copy Markdown

Important

Review skipped

Auto reviews are disabled on base/target branches other than the default branch.

Please check the settings in the CodeRabbit UI or the .coderabbit.yaml file in this repository. To trigger a single review, invoke the @coderabbitai review command.

⚙️ Run configuration

Configuration used: defaults

Review profile: CHILL

Plan: Pro

Run ID: c5d8f551-1500-4b92-bff0-51e70eb66e2a

You can disable this status message by setting the reviews.review_status to false in the CodeRabbit configuration file.

Use the checkbox below for a quick retry:

  • 🔍 Trigger review
✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests
  • Commit unit tests in branch claude/topic-extraction-benchmark-error-juhs58

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands.

The manual full-benchmark job (4 datasets x 3 seeds x k-grid x 4 models,
including BERTopic) was hitting the 45-minute timeout and getting
cancelled before finishing, not failing outright.
@nevil-mathew
nevil-mathew merged commit 2b342c2 into claude/repo-assessment-completeness-ipnli2 Jul 4, 2026
9 checks passed
@nevil-mathew
nevil-mathew deleted the claude/topic-extraction-benchmark-error-juhs58 branch July 5, 2026 04:06
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.

2 participants