traveler.core.multithread.JaxTable.from_parquet#

classmethod JaxTable.from_parquet(filename, id_col=None)[source]#

Create a dataset from a Parquet file.

Parameters:

  • filename (str or os.PathLike) – Parquet file to read.

  • id_col (str or None, optional) – Column used to initialize deterministic PRNG keys.

Returns:

Dataset populated from the Parquet columns.

Return type:

JaxDataset