Merge pull request #1298 from perib/random_state_fix

Random state fix
EpistasisLab · May 18, 2023 · 5277f10 · 5277f10
2 parents 0b45d0e + a2566ad
commit 5277f10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tpot/base.py b/tpot/base.py
@@ -1420,7 +1420,7 @@ def _compile_to_sklearn(self, expr):
         )
         sklearn_pipeline = eval(sklearn_pipeline_str, self.operators_context)
         sklearn_pipeline.memory = self._memory
-        if self.random_state:
+        if self.random_state is not None:
             # Fix random state when the operator allows
             set_param_recursive(
                 sklearn_pipeline.steps, "random_state", self.random_state