From 2efd73214406c9e52cebb7196495d9e6743e8884 Mon Sep 17 00:00:00 2001 From: Rasmus Oersoe Date: Thu, 2 May 2024 12:34:28 +0200 Subject: [PATCH] fix truth table reference --- src/graphnet/datasets/prometheus_datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/graphnet/datasets/prometheus_datasets.py b/src/graphnet/datasets/prometheus_datasets.py index 1f3435e4b..14cb2bc78 100644 --- a/src/graphnet/datasets/prometheus_datasets.py +++ b/src/graphnet/datasets/prometheus_datasets.py @@ -48,7 +48,7 @@ def _prepare_args( dataset_path = dataset_paths[0] event_nos = query_database( database=dataset_path, - query=f"SELECT event_no FROM {self._truth_table[0]}", + query=f"SELECT event_no FROM {self._truth_table}", ) train_val, test = train_test_split( event_nos["event_no"].tolist(),