Ensure we pass integer verbosity

MilesCranmer · Aug 2, 2023 · 3ff33b4 · 3ff33b4
1 parent f991f4f
commit 3ff33b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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