Conversation
There was a problem hiding this comment.
The test_skyrl_gym_generator test uses the SearchR1 dataset, but only the test split (which is much smaller than the train split), so this change enables only downloaded the test split
There was a problem hiding this comment.
This file is just a copy-paste of the original SkyRL Gym test, but left behind on test (text2sql) that needs to be made more lightweight before transferring over
There was a problem hiding this comment.
how much time does this take on L4 now?
There was a problem hiding this comment.
I measured the full block of migrated tests (the 5 files in this PR), which was originally 41min but is now 23min. Still quite slow but... a start.
There was a problem hiding this comment.
ok sg, let's revisit once all files are moved over
There was a problem hiding this comment.
Agreed, and added this as step 2 in the issue
Migrating a first batch of tests over to `gpu_ci`, for #199
2 participants
Migrating a first batch of tests over to
gpu_ci, for #199