TB: small fixes (#487)

Parallel-in-Time · Sep 20, 2024 · 2b8ebc7 · 2b8ebc7
1 parent 8420a47
commit 2b8ebc7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pySDC/core/controller.py b/pySDC/core/controller.py
@@ -47,7 +47,7 @@ def __init__(self, controller_params, description, useMPI=None):
         user_hooks = controller_params.get('hook_class', [])
         hook_classes += user_hooks if type(user_hooks) == list else [user_hooks]
         [self.add_hook(hook) for hook in hook_classes]
-        controller_params['hook_class'] = controller_params.get('hook_class', hook_classes)
+        controller_params['hook_class'] = hook_classes
 
         for hook in self.hooks:
             hook.pre_setup(step=None, level_number=None)

diff --git a/pySDC/implementations/convergence_controller_classes/interpolate_between_restarts.py b/pySDC/implementations/convergence_controller_classes/interpolate_between_restarts.py
@@ -56,7 +56,7 @@ def post_spread_processing(self, controller, step, **kwargs):
 
                 for m in range(len(level.u)):
                     level.u[m][:] = self.status.u_inter[i][m].reshape(level.prob.init[0])[:]
-                    level.f[m][:] = self.status.f_inter[i][m].reshape(level.prob.init[0])[:]
+                    level.f[m][:] = self.status.f_inter[i][m].reshape(level.f[m].shape)[:]
 
             # reset the status variables
             self.status.perform_interpolation = False