diff --git a/src/graphnet/models/detector/icecube.py b/src/graphnet/models/detector/icecube.py index c39240d7f..877a0bf65 100644 --- a/src/graphnet/models/detector/icecube.py +++ b/src/graphnet/models/detector/icecube.py @@ -51,6 +51,13 @@ def _pmt_area(self, x: torch.tensor) -> torch.tensor: class IceCubeKaggle(Detector): """`Detector` class for Kaggle Competition.""" + geometry_table_path = os.path.join( + ICECUBE_GEOMETRY_TABLE_DIR, "icecube86.parquet" + ) + xyz = ["x", "y", "z"] + string_id_column = "string" + sensor_id_column = "sensor_id" + def feature_map(self) -> Dict[str, Callable]: """Map standardization functions to each dimension of input data.""" feature_map = {