Skip to content

Commit

Permalink
lint fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
atb1995 committed Aug 14, 2024
1 parent c5d6385 commit bd76e75
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 6 deletions.
9 changes: 5 additions & 4 deletions gusto/solvers/linear_solvers.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,10 +57,11 @@ def __init__(self, equations, alpha=0.5, tau_u=0.5, tau_r=0.5, tau_t=0.5,
self.equations = equations
self.dt = equations.domain.dt
self.alpha = alpha

# Set relaxation parameters.
self.tau_u=tau_u
self.tau_t=tau_t
self.tau_r=tau_r
self.tau_u = tau_u
self.tau_t = tau_t
self.tau_r = tau_r

if solver_parameters is not None:
if not overwrite_solver_parameters:
Expand Down Expand Up @@ -145,7 +146,7 @@ class CompressibleSolver(TimesteppingSolver):
'pc_type': 'bjacobi',
'sub_pc_type': 'ilu'}}}

def __init__(self, equations, alpha=0.5, tau_u = 0.5, tau_r = 0.5, tau_t = 0.5,
def __init__(self, equations, alpha=0.5, tau_u=0.5, tau_r=0.5, tau_t=0.5,
quadrature_degree=None, solver_parameters=None,
overwrite_solver_parameters=False):
"""
Expand Down
4 changes: 2 additions & 2 deletions gusto/timestepping/semi_implicit_quasi_newton.py
Original file line number Diff line number Diff line change
Expand Up @@ -485,14 +485,14 @@ def apply(self, x_in, x_nl, x_out, label):

x_out.assign(x_in(self.field_name))
x_out += self.xF

def zero_forcing_terms(self, equation, x_in, x_out, transported_field_names):
"""
Zero forcing term F(x) for non-wind transport.
This takes x_in and x_out, where \n
x_out = x_in + scale*F(x_nl) \n
for some field x_nl and sets x_out = x_in for all non-wind transport terms
for some field x_nl and sets x_out = x_in for all non-wind transport terms
Args:
equation (:class:`PrognosticEquationSet`): the prognostic
Expand Down

0 comments on commit bd76e75

Please sign in to comment.