From b321cf04a1ce8f457324d1b04736e5ff7cfcfbf9 Mon Sep 17 00:00:00 2001 From: Julian Kuhlmann Date: Thu, 18 Jun 2026 18:07:24 +0200 Subject: [PATCH] bugfix --- icecube_data_reader/events.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/icecube_data_reader/events.py b/icecube_data_reader/events.py index cb08c4f..d670806 100644 --- a/icecube_data_reader/events.py +++ b/icecube_data_reader/events.py @@ -8,6 +8,7 @@ from pathlib import Path import numpy as np import numpy.typing as npt +from typing import Iterable from astropy.coordinates import SkyCoord from astropy import units as u from astropy.time import Time @@ -124,7 +125,7 @@ def __init__( self, energies: u.GeV, coords: SkyCoord, - types: np.ndarray, + types: Iterable[EventType], ang_errs: u.deg, mjd: Time, ): @@ -194,6 +195,7 @@ def remove(self, i: int) -> None: self._types = np.delete(self._types, i) self._ang_errs = np.delete(self._ang_errs, i) self._mjd = np.delete(self._mjd, i) + self._int_types = np.delete(self._int_types, i) def to_file( self,