Oscar Barrera

The [`_calculate_sizes`](https://github.com/graphnet-team/graphnet/blob/72c38d52fa39923bf5c6728b6ce347b224132f72/src/graphnet/data/dataset/parquet/parquet_dataset.py#L197) function (which calculates the number of events in each batch) within the `ParquetDataset` class calculates the batch size by appending the length of each file inside the batch....

ParquetDataset class missing 'labels' argument

1

In the example script `02_train_tito_model.py`, there are labels passed to the dataloader `'labels': { "direction": Direction( azimuth_key="injection_azimuth", zenith_key="injection_zenith" ) }` The SQLiteDataset handles these labels fine - but when loading...