Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Create Unittest Runner for RAG Tests, including RagQADatasets #10108

Open
wants to merge 20 commits into
base: master
Choose a base branch
from

Conversation

zaristei
Copy link
Contributor

@zaristei zaristei commented Mar 11, 2025

This PR adds a new category of tests for RAG related datasets and networks, with a runner that evaluates them with a separate set of dependencies. Right now, these tests run for:
test_web_qsp_dataset.py (also included in this PR)
test_sentence_transformer.py (currently disabled unless the full suite is enabled).

@puririshi98 puririshi98 marked this pull request as ready for review March 11, 2025 00:18
@puririshi98 puririshi98 requested a review from wsad1 as a code owner March 11, 2025 00:18
@zaristei zaristei force-pushed the zaristei/web_qsp_unittests branch from 3d1aeae to efe0db2 Compare March 11, 2025 00:32
@zaristei zaristei requested a review from rusty1s as a code owner March 11, 2025 21:04
@zaristei
Copy link
Contributor Author

zaristei commented Mar 11, 2025

I also ran the full suite on Ubuntu 22.04 CUDA 12.4 Python 10. Logs here:
pytest_output.log
TLDR:
Screenshot 2025-03-11 at 4 25 30 PM

@zaristei zaristei changed the title Base Unittests for WebQSP Create Unittest Runner for RAG Tests Mar 11, 2025
@zaristei zaristei changed the title Create Unittest Runner for RAG Tests Create Unittest Runner for RAG Tests, including RagQADatasets Mar 11, 2025
@zaristei zaristei force-pushed the zaristei/web_qsp_unittests branch from 31f9df3 to 239d479 Compare March 11, 2025 23:30
@zaristei zaristei changed the base branch from latest-txt2kg to master March 11, 2025 23:31
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