instadeepai / instadeepai/flashbax
Index trajectories at particular index from buffer
Open
enhancement
good first issue
- Dominant language
- Python
- Stars
- 283
- Forks
- 22
- PR merge metrics
- No merged PRs in 30d
Description
Can we get trajectories at particular index from the TrajectoryBuffer or ItemBuffer. Something like buffer.get_trajectory(state, trajectory_index), since buffer.sample(state, rng_key) returns a random slice from the buffer.
Contributor guide
Assessment
This issue has not been assessed yet.