Skip to content

Commit a6f680a

Browse files
author
Luigi Dello Stritto
committed
fix
1 parent 0d1cc47 commit a6f680a

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

machine_learning_hep/analysis/analyzer_jets.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -284,8 +284,8 @@ def calculate_efficiencies(self):
284284
eff_avg.Divide(project_hist(h_out, [1], {0: bins_ptjet}))
285285

286286
if eff_corr := self.cfg('efficiency.reweight'):
287-
for ipt in range(get_nbins(eff_avg, 0)):
288-
scale_bin(eff_avg, eff_corr[ipt][icat], ipt+1)
287+
for ipt in range(get_nbins(eff_avg, 0)):
288+
scale_bin(eff_avg, eff_corr[ipt][icat], ipt+1)
289289

290290
self._save_hist(eff_avg, f'eff/h_pthf_effnew_{cat}.png')
291291
self.h_effnew_pthf[cat] = eff_avg

machine_learning_hep/processer.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -429,13 +429,16 @@ def dfuse(df_spec):
429429
self.logger.info('merging %s with %s on %s into %s', base, ref, on, out)
430430
if not is_numeric_dtype(dfs[base][on]):
431431
self.logger.info('exploding dataframe %s on variable %s', base, on)
432-
dfs[out] = dfmerge(dfs[base].explode(on), dfs[ref], left_on=['df', on], suffixes=(f'_{base}', None), right_index=True)
432+
dfs[out] = dfmerge(
433+
exploded_df, dfs[ref], left_on=['df', on], suffixes=(f'_{base}', None), right_index=True)
433434
else:
434-
dfs[out] = dfmerge(dfs[base], dfs[ref], left_on=['df', on], suffixes=(f'_{base}', None), right_index=True)
435+
dfs[out] = dfmerge(
436+
dfs[base], dfs[ref], left_on=['df', on], suffixes=(f'_{base}', None), right_index=True)
435437
else:
436438
var = self.df_read[ref]['index']
437439
self.logger.info('merging %s with %s on %s (default) into %s', base, ref, var, out)
438-
dfs[out] = dfmerge(dfs[base], dfs[ref], left_on=['df', var], suffixes=(f'_{base}', None), right_index=True)
440+
dfs[out] = dfmerge(
441+
dfs[base], dfs[ref], left_on=['df', var], suffixes=(f'_{base}', None), right_index=True)
439442
if 'extra' in m_spec:
440443
self.logger.debug(' %s -> extra', out)
441444
for col_name, col_val in m_spec['extra'].items():

0 commit comments

Comments
 (0)