diff --git a/km3io/__init__.py b/km3io/__init__.py index 52ba6348e74fb7334c04ebe5a1a1b1025869c31a..1d7af3825e9df7cfb878432cd64ec60197e94f20 100644 --- a/km3io/__init__.py +++ b/km3io/__init__.py @@ -5,4 +5,3 @@ version = get_distribution(__name__).version from .offline import OfflineReader from .online import OnlineReader from .gseagen import GSGReader -from . import patches diff --git a/km3io/patches.py b/km3io/patches.py deleted file mode 100644 index 8d3be710146cb7db031f7e7e64578d924fa2b0be..0000000000000000000000000000000000000000 --- a/km3io/patches.py +++ /dev/null @@ -1,17 +0,0 @@ -import awkward0 as ak0 -import awkward as ak1 - -# to avoid infinite recursion -old_getitem = ak0.ChunkedArray.__getitem__ - - -def new_getitem(self, item): - """Monkey patch the getitem in awkward.ChunkedArray to apply - awkward1.Array masks on awkward.ChunkedArray""" - if isinstance(item, (ak1.Array, ak0.ChunkedArray)): - return ak1.Array(self)[item] - else: - return old_getitem(self, item) - - -ak0.ChunkedArray.__getitem__ = new_getitem