graphnet.data.utilities.random module

RNG-related utility functions relevant to the graphnet.data package.

graphnet.data.utilities.random.pairwise_shuffle(i3_list, gcd_list)

Shuffle the I3 file list and the correponding gcd file list.

This is handy because it ensures a more even extraction load for each worker.

Parameters:

• files_list – List of I3 file paths.

• gcd_list (List[str]) – List of corresponding gcd file paths.

• i3_list (List[str])

Returns:

List of shuffled I3 file paths. gcd_shuffled: List of corresponding gcd file paths.

Return type:

i3_shuffled