Merge pull request graphnet-team#704 from RasmusOrsoe/main

fix truth table reference
RasmusOrsoe · May 2, 2024 · 448a264 · 448a264
2 parents 0ac4410 + 2efd732
commit 448a264
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(),