graphnet.data.readers.prometheus_reader module

Modules for reading data files from the Prometheus project.

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

Bases: GraphNeTFileReader

A class for reading parquet files from Prometheus simulation.

Construct Logger.

Parameters:

  • name (str | None)

  • class_name (str | None)

  • level (int)

  • log_folder (str | None)

  • kwargs (Any)

find_files(path)[source]

Search folder(s) for parquet files.

Parameters:

path (Union[str, List[str]]) – directory to search for parquet files.

Return type:

List[str]

Returns:

List of parquet files in the folders.