| """ | |
| NBA graph dataloader for MoFlow-Graph. | |
| The existing seq_collate_nba already includes 'past_traj_original_scale' | |
| which is everything the FutureInteractionGraph needs to reconstruct | |
| inter-agent absolute positions. This module re-exports the existing | |
| dataset + collate unchanged; the only new export is seq_collate_nba_graph | |
| which is an alias kept for clarity in fm_nba_graph.py. | |
| Graph-topology construction (edge_index) is handled inside | |
| FutureInteractionGraph.forward() using pre-registered buffers — so no | |
| changes to the collate function or dataset are required. | |
| """ | |
| from data.dataloader_nba import ( | |
| NBADatasetMinMax, | |
| seq_collate_nba as seq_collate_nba_graph, # alias — identical behaviour | |
| seq_collate_imle_train, | |
| ) | |
| __all__ = [ | |
| 'NBADatasetMinMax', | |
| 'seq_collate_nba_graph', | |
| 'seq_collate_imle_train', | |
| ] | |