Generate a list of experiences for testing.
(task_num: int, repeat_times: int = 1, step_num: int = 1)
| 19 | |
| 20 | |
| 21 | def get_experiences(task_num: int, repeat_times: int = 1, step_num: int = 1) -> List[Experience]: |
| 22 | """Generate a list of experiences for testing.""" |
| 23 | return [ |
| 24 | Experience( |
| 25 | eid=EID(task=i, run=j, step=k), |
| 26 | tokens=torch.zeros((5,)), |
| 27 | prompt_length=4, |
| 28 | reward=j, |
| 29 | logprobs=torch.tensor([0.1]), |
| 30 | ) |
| 31 | for i in range(task_num) |
| 32 | for j in range(repeat_times) |
| 33 | for k in range(step_num) |
| 34 | ] |
| 35 | |
| 36 | |
| 37 | class TestExperiencePipeline(RayUnittestBaseAsync): |
no test coverage detected