"""
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',
]