clip rescale factor to avoid overflow warning (#10)

spatial-model-editor · Aug 25, 2021 · df22b9c · df22b9c
1 parent 00443f5
commit df22b9c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sme_contrib/optimize.py b/sme_contrib/optimize.py
@@ -327,7 +327,7 @@ def _get_dcdt(self, result):
     def _rescale(self, result):
         c = self._get_conc(result)
         dcdt = self._get_dcdt(result)
-        scale_factor = 1.0 / np.amax(c)
+        scale_factor = 1.0 / np.clip(np.amax(c), 1e-100, 1e100)
         return scale_factor * c, scale_factor * dcdt
 
     def _obj_func(self, params, verbose=False):