graphnet.data.readers.internal_parquet_reader module

Module containing different reader for GraphNeT internal parquet format.

class graphnet.data.readers.internal_parquet_reader.ParquetReader(name, class_name, level, log_folder, **kwargs)[source]

Bases: GraphNeTFileReader

A class for reading the internal GraphNeT parquet format.

Construct Logger.

Parameters:

  • name (str | None)

  • class_name (str | None)

  • level (int)

  • log_folder (str | None)

  • kwargs (Any)

find_files(path)[source]

Search parquet folders for files.

Parameters:

path (Union[str, List[str]]) – directory containing the parquet folders.

Return type:

List[str]

Returns:

List of parquet files in the folders.