Skip to content

Commit

Permalink
allow TSEval run_routine to take generic signal name like WIMP40 and …
Browse files Browse the repository at this point in the history
…get the source for all components
  • Loading branch information
Makayla Trask committed Mar 12, 2024
1 parent b80355b commit cd8eae2
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions flamedisx/likelihood.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,6 +128,7 @@ def __init__(
if isinstance(data, pd.DataFrame) or data is None:
# Only one dataset
data = {DEFAULT_DSETNAME: data}

if not isinstance(list(sources.values())[0], dict):
# Sources only specified for one dataset
assert len(data) == 1, "Specify which sources belong to which dataset"
Expand Down Expand Up @@ -272,6 +273,8 @@ def set_data(self,
for s in self.sources.values():
s.set_data(None)
return

print('before batch info line 276')

batch_info = np.zeros((len(self.dsetnames), 3), dtype=int)

Expand Down

0 comments on commit cd8eae2

Please sign in to comment.