random

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

graphnet.data.utilities.random.pairwise_shuffle(i3_list, gcd_list)[source]

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