tinker_cookbook.rl.assemble_training_data

tinker_cookbook.rl.assemble_training_data( trajectory_groups_P, advantages_P)

Convert trajectories to training data format.

Parameters:

• trajectory_groups_P ( list[ TrajectoryGroup ]) – Groups of trajectories to convert into training datums.
• advantages_P ( list[torch.Tensor]) – Per-group advantage tensors, one per trajectory group, as returned by compute_advantages.

Returns: tuple[list[tinker. Datum ], list[dict[str, int]]] – A flat list of training datums and a parallel list of metadata dicts mapping each datum back to its group_idx and traj_idx.