From bc5ff2b1771b23c2da89a4cc73b01aef175d1b32 Mon Sep 17 00:00:00 2001 From: Doug Branton Date: Fri, 22 Dec 2023 11:51:47 -0800 Subject: [PATCH] pair down rng logic --- src/tape/ensemble.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py index 9ef50499..e642263a 100644 --- a/src/tape/ensemble.py +++ b/src/tape/ensemble.py @@ -1870,10 +1870,8 @@ def select_random_timeseries(self, seed=None): in larger partitions. """ - if seed is not None: - rng = np.random.default_rng(seed) - else: - rng = np.random + + rng = np.random.default_rng(seed) # We will select one partition at random to select an object from partitions = np.array(range(self.object.npartitions))