tensorflow.python.framework.errors_impl.InvalidArgumentError: cannot compute Mul as input #1(zero-based) was expected to be a float tensor but is a double tensor [Op:Mul] name: 
Run `reticulate::py_last_error()` for details.

9.
stop(structure(list(message = "tensorflow.python.framework.errors_impl.InvalidArgumentError: cannot compute Mul as input #1(zero-based) was expected to be a float tensor but is a double tensor [Op:Mul] name: \n\033[90mRun \033]8;;rstudio:run:reticulate::py_last_error()\a`reticulate::py_last_error()`\033]8;;\a for details.\033[39m", 
call = py_call_impl(callable, call_args$unnamed, call_args$named)), class = c("tensorflow.python.framework.errors_impl.InvalidArgumentError", 
"tensorflow.python.framework.errors_impl.OpError", "python.builtin.Exception", 
"python.builtin.BaseException", "python.builtin.object", "error", ...

8.
raise_from_not_ok_status at ops.py#5983

7.
error_handler at traceback_utils.py#153

6.
compute_metrics at flow_matching.py#117

5.
compute_metrics at continuous_approximator.py#95

4.
build_from_data at approximator.py#23

3.
fit at approximator.py#82

2.
fit at continuous_approximator.py#109

1.
approximator$fit(epochs = 3L, dataset = training_dataset, validation_data = validation_dataset)```