fix tensorflow incompatible types

bayesflow-org · Feb 7, 2025 · f135ecd · f135ecd
1 parent 2c5f872
commit f135ecd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bayesflow/utils/integrate.py b/bayesflow/utils/integrate.py
@@ -199,7 +199,7 @@ def body(_state, _time, _step_size, _step):
         time_remaining = stop_time - (_time + _step_size)
 
         min_step_size = time_remaining / (max_steps - _step)
-        max_step_size = time_remaining / keras.ops.maximum(min_steps - _step, 1)
+        max_step_size = time_remaining / keras.ops.maximum(min_steps - _step, 1.0)
 
         # reorder
         min_step_size, max_step_size = (