Skip to content

Commit 678dea1

Browse files
committed
pep8 fixes
1 parent b89f838 commit 678dea1

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

pySDC/projects/matrixPFASST/allinclusive_matrix_nonMPI.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -72,10 +72,12 @@ def __init__(self, num_procs, controller_params, description):
7272
N = np.zeros((self.nnodes, self.nnodes))
7373
N[:, -1] = 1
7474

75-
self.C = np.array(np.eye(self.nsteps * self.nnodes * self.nspace) - \
76-
self.dt * np.kron(np.eye(self.nsteps), np.kron(Q, A)) - np.kron(E, np.kron(N, np.eye(self.nspace))))
77-
self.P = np.array(np.eye(self.nsteps * self.nnodes * self.nspace) - \
78-
self.dt * np.kron(np.eye(self.nsteps), np.kron(Qd, A)))
75+
self.C = np.eye(self.nsteps * self.nnodes * self.nspace) - \
76+
self.dt * np.kron(np.eye(self.nsteps), np.kron(Q, A)) - np.kron(E, np.kron(N, np.eye(self.nspace)))
77+
self.C = np.array(self.C)
78+
self.P = np.eye(self.nsteps * self.nnodes * self.nspace) - \
79+
self.dt * np.kron(np.eye(self.nsteps), np.kron(Qd, A))
80+
self.P = np.array(self.P)
7981

8082
if self.nlevels > 1:
8183
prob_c = self.MS[0].levels[1].prob
@@ -96,9 +98,10 @@ def __init__(self, num_procs, controller_params, description):
9698
self.Tcf = np.array(np.kron(np.eye(self.nsteps), np.kron(TcfQ, TcfA)))
9799
self.Tfc = np.array(np.kron(np.eye(self.nsteps), np.kron(TfcQ, TfcA)))
98100

99-
self.Pc = np.array(np.eye(self.nsteps * nnodesc * self.nspace_c) - \
101+
self.Pc = np.eye(self.nsteps * nnodesc * self.nspace_c) - \
100102
self.dt * np.kron(np.eye(self.nsteps), np.kron(Qdc, Ac)) - \
101-
np.kron(E, np.kron(Nc, np.eye(self.nspace_c))))
103+
np.kron(E, np.kron(Nc, np.eye(self.nspace_c)))
104+
self.Pc = np.array(self.Pc)
102105

103106
self.u = np.zeros(self.nsteps * self.nnodes * self.nspace)
104107
self.res = np.zeros(self.nsteps * self.nnodes * self.nspace)

0 commit comments

Comments
 (0)