Skip to content

Commit 7c16a14

Browse files
committed
Moving named tuples definition to top of module utils.py
1 parent e6b5203 commit 7c16a14

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

pydmd/utils.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,12 @@
77
import numpy as np
88
from numpy.lib.stride_tricks import sliding_window_view
99

10+
# Named tuples used functions.
11+
# compute_svd uses "SVD",
12+
# compute_tlsq uses "TLSQ".
13+
SVD = namedtuple("SVD", ["U", "s", "V"])
14+
TLSQ = namedtuple("TLSQ", ["X_denoised", "Y_denoised"])
15+
1016

1117
def _svht(sigma_svd: np.ndarray, rows: int, cols: int) -> int:
1218
"""
@@ -140,7 +146,7 @@ def compute_tlsq(
140146
V = np.linalg.svd(np.append(X, Y, axis=0), full_matrices=False)[-1]
141147
rank = min(tlsq_rank, V.shape[0])
142148
VV = V[:rank, :].conj().T.dot(V[:rank, :])
143-
TLSQ = namedtuple("TLSQ", ["X_denoised", "Y_denoised"])
149+
144150
return TLSQ(X.dot(VV), Y.dot(VV))
145151

146152

@@ -179,7 +185,6 @@ def compute_svd(
179185
U = U[:, :rank]
180186
V = V[:, :rank]
181187
s = s[:rank]
182-
SVD = namedtuple("SVD", ["U", "s", "V"])
183188

184189
return SVD(U, s, V)
185190

0 commit comments

Comments
 (0)