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

Enable test_aegis_classifier and test_instruction_data_guard_classifier #456

Open
sarahyurick opened this issue Dec 23, 2024 · 0 comments
Open
Assignees

Comments

@sarahyurick
Copy link
Collaborator

#421 added PyTests for all of our supported classifiers. However, we are currently skipping the tests for the Aegis and Instruction-Data-Guard classifiers (which require the user to pass a Hugging Face token) until we download and cache the models to our self-hosted GPU runner. This method is preferred over passing a Hugging Face token into the tests because we do not want to get rate limited by Hugging Face.

We should be able to enable these tests after I am able to add the models to .cache/huggingface/hub (wherever the Hugging Face cache directory is stored on our runner).

@sarahyurick sarahyurick self-assigned this Dec 23, 2024
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

No branches or pull requests

1 participant