internal_sqlite_reader

Reader for GraphNeT internal SQLite format.

class graphnet.data.readers.internal_sqlite_reader.SQLiteReader(subset_size)[source]

Bases: GraphNeTFileReader

A class for reading the internal GraphNeT SQLite format (.db).

Construct SQLiteReader.

Parameters:

subset_size (int, default: 10000) – Number of events per fileset chunk. Defaults to 10000.

find_files(path)[source]

Produce filesets of (database path, subset of event numbers).

Parameters:

path (Union[str, List[str]]) – Directory or list of directories containing .db files, or a single .db file path.

Return type:

List[SQLiteFileSet]

Returns:

List of SQLiteFileSet objects.