pair down rng logic

lincc-frameworks · Dec 22, 2023 · bc5ff2b · bc5ff2b
1 parent f8c5847
commit bc5ff2b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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))