From 0c6ccdf154f38004ff324b7f3462d58a8632f2e0 Mon Sep 17 00:00:00 2001 From: Valentin Boettcher Date: Sun, 14 Aug 2022 15:53:15 +0200 Subject: [PATCH] fix slice --- hopsflow/util.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/hopsflow/util.py b/hopsflow/util.py index 5ca928e..24e06ba 100644 --- a/hopsflow/util.py +++ b/hopsflow/util.py @@ -116,9 +116,7 @@ class EnsembleValue: def __getitem__(self, index): return EnsembleValue(self._value[index]) - def slice(self, *args): - slc = slice(*args) - + def slice(self, slc: Union[np.ndarray, slice]): results = [] for N, val, σ in self.aggregate_iterator: results.append((N, val[slc], σ[slc]))