Hi,
I want to load a OXE dataset (like RT-1), filter the episodes that accomplish a specific task, and save the filtered episodes in the same tfrecord format as the original OXE dataset. Could you help give a minimal example code of how to do this? Thanks!