Adds a move-to-device in pr replication (#61)

Author: miguelgondu

src/poli_baselines/core/utils/bo_pr/run_one_replication.py

@@ -540,7 +540,7 @@ def callback(i, loss, grad, X):
             gc.collect()
 
         # Get the new observations and update the data.
-        new_y = eval_problem(candidates, base_function=base_function)
+        new_y = eval_problem(candidates, base_function=base_function).to(**tkwargs)
 
         if use_trust_region:
             old_length = trbo_state.length