Skip to content

Minor fix in datacard functions #465

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ywan2 wants to merge 3 commits into
base: master
Choose a base branch
from
Open

Minor fix in datacard functions #465

Changes from 2 commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
37fc6f6
update scalings function in datacard_tools.py
Mar 21, 2025
4d79d98
fix scalings and asimov datacard
Jun 3, 2025
f659568
fix indent
Jun 3, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 13 additions & 7 deletions topeft/modules/datacard_tools.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -857,6 +857,14 @@ def analyze(self,km_dist,ch,selected_wcs, crop_negative_bins, wcs_dict):
raise RuntimeError("filling obs data more than once!")
for sp_key,arr in data_sm.items():
data_obs += arr
if not self.use_real_data and wcs_dict is not None:
# Create asimov dataset, 'vals' are WCs specified at the command-line
vals = wcs_dict
decomposed_templates = self.decompose(proc_hist,proc_sumw2,wcs,vals)
data_sm = decomposed_templates.pop("sm")
for sp_key,arr in data_sm.items():
if sp_key.systematic == "nominal":
data_obs += arr
if not self.use_AAC:
decomposed_templates = {k: v for k, v in decomposed_templates.items() if k == 'sm'}
for base,v in decomposed_templates.items():
Expand Down Expand Up @@ -895,12 +903,9 @@ def analyze(self,km_dist,ch,selected_wcs, crop_negative_bins, wcs_dict):
if syst == "nominal" and base == "sm":
if self.verbose:
print(f"\t{proc_name:<12}: {sum_arr:.4f} {arr[0]}")
if not self.use_real_data:
if not self.use_real_data and wcs_dict is None:
# Create asimov dataset
vals = wcs_dict # set wcs to certain values from command line
decomposed_templates_Asimov = self.decompose(proc_hist,proc_sumw2,wcs,vals)
data_sm = decomposed_templates_Asimov.pop("sm")
data_obs += data_sm[sp_key]
data_obs += arr
if syst == "nominal":
hist_name = f"{proc_name}"
text_card_info[proc_name]["rate"] = sum_arr
Expand Down Expand Up @@ -963,10 +968,10 @@ def analyze(self,km_dist,ch,selected_wcs, crop_negative_bins, wcs_dict):
# obtain the scalings for scalings.json file
if p in self.SIGNALS:
if self.wc_scalings:
scalings = h[{'channel':ch,'process':p,'systematic':'nominal'}].make_scaling(self.wc_scalings)
scalings = h[{'channel':ch,'process':p,'systematic':'nominal'}].make_scaling(flow='show', wc_list=self.wc_scalings)
self.scalings_json = self.make_scalings_json(self.scalings,ch,km_dist,p,self.wc_scalings,scalings)
else:
scalings = h[{'channel':ch,'process':p,'systematic':'nominal'}].make_scaling()
scalings = h[{'channel':ch,'process':p,'systematic':'nominal'}].make_scaling(flow='show')
self.scalings_json = self.make_scalings_json(self.scalings,ch,km_dist,p,h.wc_names,scalings)
f["data_obs"] = to_hist(data_obs,"data_obs")

Expand Down Expand Up @@ -1100,6 +1105,7 @@ def analyze(self,km_dist,ch,selected_wcs, crop_negative_bins, wcs_dict):
f.write("* autoMCStats 10\n")
else:
f.write("* autoMCStats -1\n")

dt = time.time() - tic
print(f"File Write Time: {dt:.2f} s")
print(f"Total Hists Written: {num_h}")
Expand Down
Loading