fix reproducibility bug

EpistasisLab · Sep 12, 2024 · bcc890e · bcc890e
1 parent 923782a
commit bcc890e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tpot2/search_spaces/pipelines/choice.py b/tpot2/search_spaces/pipelines/choice.py
@@ -12,7 +12,7 @@ def __init__(self, search_spaces : List[SklearnIndividualGenerator], rng=None) -
         super().__init__()
 
         self.search_spaces = search_spaces
-        self.node = np.random.default_rng(rng).choice(self.search_spaces).generate()
+        self.node = np.random.default_rng(rng).choice(self.search_spaces).generate(rng=rng)
 
 
     def mutate(self, rng=None):
@@ -23,7 +23,7 @@ def mutate(self, rng=None):
             return self._mutate_node(rng)
 
     def _mutate_select_new_node(self, rng=None):
-        self.node = random.choice(self.search_spaces).generate()
+        self.node = random.choice(self.search_spaces).generate(rng=rng)
         return True
 
     def _mutate_node(self, rng=None):