MilesCranmer commited on
Commit
3ff33b4
1 Parent(s): f991f4f

Ensure we pass integer verbosity

Browse files
Files changed (1) hide show
  1. pysr/sr.py +2 -2
pysr/sr.py CHANGED
@@ -1313,7 +1313,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1313
  "constraints": {},
1314
  "multithreading": self.procs != 0 and self.cluster_manager is None,
1315
  "batch_size": 1,
1316
- "update_verbosity": self.verbosity,
1317
  "progress": buffer_available,
1318
  }
1319
  packed_modified_params = {}
@@ -1805,7 +1805,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1805
  return_state=True,
1806
  addprocs_function=cluster_manager,
1807
  progress=progress and self.verbosity > 0 and len(y.shape) == 1,
1808
- verbosity=self.verbosity,
1809
  )
1810
 
1811
  # Set attributes
 
1313
  "constraints": {},
1314
  "multithreading": self.procs != 0 and self.cluster_manager is None,
1315
  "batch_size": 1,
1316
+ "update_verbosity": int(self.verbosity),
1317
  "progress": buffer_available,
1318
  }
1319
  packed_modified_params = {}
 
1805
  return_state=True,
1806
  addprocs_function=cluster_manager,
1807
  progress=progress and self.verbosity > 0 and len(y.shape) == 1,
1808
+ verbosity=int(self.verbosity),
1809
  )
1810
 
1811
  # Set attributes