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.

Sign up for GitHub

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

Jump to bottom

Mock model loading in test_run_locally #18

Open
viswavi opened this issue Apr 21, 2023 · 1 comment
Open

Mock model loading in test_run_locally #18

viswavi opened this issue Apr 21, 2023 · 1 comment
Assignees
@viswavi

Comments

@viswavi
Copy link
Collaborator

viswavi commented Apr 21, 2023

In our integration test, we're currently loading a model in the Trainer. this should be mocked to avoid having to download a full transformers model each time this test is run on a new machine.
@viswavi viswavi self-assigned this Apr 21, 2023
@zhaochenyang20
Copy link
Collaborator

And apart from this, my unit test for SimpleDatasetGenerator also makes the unit test slow. (And somehow costly)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@viswavi viswavi

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@viswavi @zhaochenyang20