Move ExhaustPlugin to strax (#59)

axidence/plugin.py

@@ -6,28 +6,9 @@
 import straxen
 from straxen import units
 
-
 export, __all__ = strax.exporter()
 
 
-@export
-class ExhaustPlugin(Plugin):
-    """Plugin that exhausts all chunks when fetching data."""
-
-    def _fetch_chunk(self, d, iters, check_end_not_before=None):
-        while super()._fetch_chunk(d, iters, check_end_not_before=check_end_not_before):
-            pass
-        return False
-
-    def do_compute(self, chunk_i=None, **kwargs):
-        if chunk_i != 0:
-            raise RuntimeError(
-                f"{self.__class__.__name__} is an ExhaustPlugin. "
-                "It should read all chunks together can process them together."
-            )
-        return super().do_compute(chunk_i=chunk_i, **kwargs)
-
-
 @export
 class RunMetaPlugin(Plugin):
     """Plugin that provides run metadata."""

axidence/plugins/meta/run_meta.py

@@ -1,7 +1,6 @@
 import numpy as np
 import strax
-
-from ...plugin import ExhaustPlugin
+from strax import ExhaustPlugin
 
 
 class RunMeta(ExhaustPlugin):

axidence/plugins/pairing/peaks_paired.py

@@ -2,7 +2,7 @@
 from immutabledict import immutabledict
 import numpy as np
 import strax
-from strax import Plugin, DownChunkingPlugin
+from strax import Plugin, ExhaustPlugin, DownChunkingPlugin
 import straxen
 from straxen import units
 from straxen import PeakProximity
@@ -11,7 +11,6 @@
 from ...utils import copy_dtype
 from ...samplers import SAMPLERS
 from ...dtypes import peak_positions_dtype
-from ...plugin import ExhaustPlugin
 
 
 class PeaksPaired(ExhaustPlugin, DownChunkingPlugin):

axidence/plugins/salting/event_building.py

@@ -2,11 +2,11 @@
 from typing import Tuple
 import numpy as np
 import strax
+from strax import ExhaustPlugin
 import straxen
 from straxen import Events, EventBasics
 
 from ...utils import needed_dtype, merge_salted_real
-from ...plugin import ExhaustPlugin
 
 
 class EventsSalted(Events, ExhaustPlugin):

axidence/plugins/salting/events_salting.py

@@ -1,12 +1,11 @@
 import numpy as np
 import strax
-from strax import DownChunkingPlugin
+from strax import ExhaustPlugin, DownChunkingPlugin
 import straxen
 from straxen import units, EventBasics, EventPositions
 
 from ...utils import copy_dtype
 from ...samplers import SAMPLERS
-from ...plugin import ExhaustPlugin
 
 
 class EventsSalting(ExhaustPlugin, DownChunkingPlugin, EventPositions, EventBasics):

pyproject.toml

@@ -22,7 +22,7 @@ repository = "https://github.com/dachengx/axidence"
 documentation = "https://readthedocs.org/projects/axidence/"
 
 [tool.poetry.dependencies]
-strax = ">=1.6.2"
+strax = ">=1.6.3"
 straxen = ">=2.2.1"
 GOFevaluation = ">=0.1.4"