Skip to content

Commit 6188313

Browse files
Update RP_maker.py
1 parent 92395a7 commit 6188313

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

SMdRQA/RP_maker.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,8 @@ def RP_computer(
144144
Marr = []
145145
EPS = []
146146
BOUND = []
147+
148+
147149

148150
for File in tqdm(files):
149151
try:
@@ -222,6 +224,9 @@ def RP_computer(
222224
Marr = []
223225
EPS = []
224226
BOUND = []
227+
U_arr = []
228+
N_arr = []
229+
D_arr = []
225230

226231
for File in tqdm(files):
227232
try:
@@ -235,7 +240,9 @@ def RP_computer(
235240
n = M
236241
d = N
237242
u = data
238-
243+
U_arr.append(u)
244+
N_arr.append(n)
245+
D_arr.append(d)
239246
sd = 3 * np.std(u)
240247
print('starting tau calculation ...')
241248
tau = findtau(u, n, d, 0)
@@ -312,7 +319,7 @@ def RP_computer(
312319
m_ = Marr[i2]
313320
#
314321
if 'eps' in group_level_estimates:
315-
eps_ = np.mean(EPS)
322+
eps_ = findeps_multi(U_arr, N_arr, D_arr, Marr, TAU, reqrr, rr_delta, epsmin, epsmax, epsdiv)
316323
else:
317324
eps_ = EPS[i2]
318325
#

0 commit comments

Comments
 (0)