diff --git a/pysr/sr.py b/pysr/sr.py index c2c8a0f1f..383abf636 100644 --- a/pysr/sr.py +++ b/pysr/sr.py @@ -1313,7 +1313,7 @@ def _validate_and_set_init_params(self): "constraints": {}, "multithreading": self.procs != 0 and self.cluster_manager is None, "batch_size": 1, - "update_verbosity": self.verbosity, + "update_verbosity": int(self.verbosity), "progress": buffer_available, } packed_modified_params = {} @@ -1805,7 +1805,7 @@ def _run(self, X, y, mutated_params, weights, seed): return_state=True, addprocs_function=cluster_manager, progress=progress and self.verbosity > 0 and len(y.shape) == 1, - verbosity=self.verbosity, + verbosity=int(self.verbosity), ) # Set attributes