Skip to content

Commit 2be53b8

Browse files
author
Thomas Baumann
committed
Adhering to python standards for naming classes
1 parent 6191177 commit 2be53b8

File tree

6 files changed

+16
-17
lines changed

6 files changed

+16
-17
lines changed

pySDC/core/Controller.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from pySDC.core.BaseTransfer import base_transfer
77
from pySDC.helpers.pysdc_helper import FrozenClass
88
from pySDC.implementations.convergence_controller_classes.check_convergence import CheckConvergence
9-
from pySDC.implementations.hooks.default_hook import default_hooks
9+
from pySDC.implementations.hooks.default_hook import DefaultHooks
1010

1111

1212
# short helper class to add params as attributes
@@ -42,7 +42,7 @@ def __init__(self, controller_params, description):
4242

4343
# check if we have a hook on this list. If not, use default class.
4444
self.__hooks = []
45-
hook_classes = [default_hooks]
45+
hook_classes = [DefaultHooks]
4646
user_hooks = controller_params.get('hook_class', [])
4747
hook_classes += user_hooks if type(user_hooks) == list else [user_hooks]
4848
[self.add_hook(hook) for hook in hook_classes]

pySDC/implementations/convergence_controller_classes/estimate_embedded_error.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from pySDC.core.ConvergenceController import ConvergenceController, Pars
44
from pySDC.implementations.convergence_controller_classes.store_uold import StoreUOld
5-
from pySDC.implementations.hooks.log_embedded_error_estimate import log_embedded_error_estimate
5+
from pySDC.implementations.hooks.log_embedded_error_estimate import LogEmbeddedErrorEstimate
66

77
from pySDC.implementations.sweeper_classes.Runge_Kutta import RungeKutta
88

@@ -26,7 +26,7 @@ def __init__(self, controller, params, description, **kwargs):
2626
"""
2727
super(EstimateEmbeddedError, self).__init__(controller, params, description, **kwargs)
2828
self.buffers = Pars({'e_em_last': 0.0})
29-
controller.add_hook(log_embedded_error_estimate)
29+
controller.add_hook(LogEmbeddedErrorEstimate)
3030

3131
@classmethod
3232
def get_implementation(cls, flavor):

pySDC/implementations/convergence_controller_classes/estimate_extrapolation_error.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from pySDC.core.ConvergenceController import ConvergenceController, Status
55
from pySDC.core.Errors import DataError
66
from pySDC.implementations.datatype_classes.mesh import mesh, imex_mesh
7-
from pySDC.implementations.hooks.log_extrapolated_error_estimate import log_extrapolated_error_estimate
7+
from pySDC.implementations.hooks.log_extrapolated_error_estimate import LogExtrapolationErrorEstimate
88

99

1010
class EstimateExtrapolationErrorBase(ConvergenceController):
@@ -28,7 +28,7 @@ def __init__(self, controller, params, description, **kwargs):
2828
self.prev = Status(["t", "u", "f", "dt"]) # store solutions etc. of previous steps here
2929
self.coeff = Status(["u", "f", "prefactor"]) # store coefficients for extrapolation here
3030
super(EstimateExtrapolationErrorBase, self).__init__(controller, params, description)
31-
controller.add_hook(log_extrapolated_error_estimate)
31+
controller.add_hook(LogExtrapolationErrorEstimate)
3232

3333
def setup(self, controller, params, description, **kwargs):
3434
"""
@@ -254,7 +254,7 @@ class EstimateExtrapolationErrorNonMPI(EstimateExtrapolationErrorBase):
254254

255255
def setup(self, controller, params, description, **kwargs):
256256
"""
257-
Add a no parameter 'no_storage' which decides whether the standart or the no-memory-overhead version is run,
257+
Add a no parameter 'no_storage' which decides whether the standard or the no-memory-overhead version is run,
258258
where only values are used for extrapolation which are in memory of other processes
259259
260260
Args:

pySDC/implementations/hooks/default_hook.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from pySDC.core.Hooks import hooks
33

44

5-
class default_hooks(hooks):
5+
class DefaultHooks(hooks):
66
"""
77
Hook class to contain the functions called during the controller runs (e.g. for calling user-routines)
88
@@ -27,7 +27,7 @@ class default_hooks(hooks):
2727
"""
2828

2929
def __init__(self):
30-
super(default_hooks, self).__init__()
30+
super(DefaultHooks, self).__init__()
3131
self.__t0_setup = None
3232
self.__t0_run = None
3333
self.__t0_predict = None

pySDC/projects/Resilience/hook.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
from pySDC.core.Hooks import hooks
2-
from pySDC.implementations.hooks.log_solution import log_solution
3-
from pySDC.implementations.hooks.log_embedded_error_estimate import log_embedded_error_estimate
4-
from pySDC.implementations.hooks.log_extrapolated_error_estimate import log_extrapolated_error_estimate
2+
from pySDC.implementations.hooks.log_solution import LogSolution
3+
from pySDC.implementations.hooks.log_embedded_error_estimate import LogEmbeddedErrorEstimate
4+
from pySDC.implementations.hooks.log_extrapolated_error_estimate import LogExtrapolationErrorEstimate
55

66

7-
hook_collection = [log_solution, log_embedded_error_estimate, log_extrapolated_error_estimate]
7+
hook_collection = [LogSolution, LogEmbeddedErrorEstimate, LogExtrapolationErrorEstimate]
88

99

1010
class log_data(hooks):

pySDC/projects/Resilience/vdp.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,7 @@
88
from pySDC.implementations.controller_classes.controller_nonMPI import controller_nonMPI
99
from pySDC.implementations.convergence_controller_classes.adaptivity import Adaptivity
1010
from pySDC.core.Errors import ProblemError
11-
from pySDC.projects.Resilience.hook import log_data
12-
from pySDC.implementations.hooks.log_solution import log_solution
11+
from pySDC.projects.Resilience.hook import log_data, hook_collection
1312

1413

1514
def plot_step_sizes(stats, ax):
@@ -29,7 +28,7 @@ def plot_step_sizes(stats, ax):
2928
p = np.array([me[1][1] for me in get_sorted(stats, type='u', recomputed=False, sortby='time')])
3029
t = np.array([me[0] for me in get_sorted(stats, type='u', recomputed=False, sortby='time')])
3130

32-
e_em = np.array(get_sorted(stats, type='e_embedded', recomputed=False, sortby='time'))[:, 1]
31+
e_em = np.array(get_sorted(stats, type='error_embedded_estimate', recomputed=False, sortby='time'))[:, 1]
3332
dt = np.array(get_sorted(stats, type='dt', recomputed=False, sortby='time'))
3433
restart = np.array(get_sorted(stats, type='restart', recomputed=None, sortby='time'))
3534

@@ -139,7 +138,7 @@ def run_vdp(
139138
# initialize controller parameters
140139
controller_params = dict()
141140
controller_params['logger_level'] = 30
142-
controller_params['hook_class'] = [hook_class, log_solution]
141+
controller_params['hook_class'] = hook_collection + [hook_class]
143142
controller_params['mssdc_jac'] = False
144143

145144
if custom_controller_params is not None:

0 commit comments

Comments
 (0)