File size: 841 Bytes
0d52562 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | """
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',
]
|