Skip to content

Commit

Permalink
style: black
Browse files Browse the repository at this point in the history
Signed-off-by: F.N. Claessen <[email protected]>
  • Loading branch information
Flix6x committed Dec 30, 2023
1 parent e956b65 commit 6531e09
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion timely_beliefs/beliefs/classes.py
Original file line number Diff line number Diff line change
Expand Up @@ -615,7 +615,9 @@ def apply_belief_timing_filters(q):
# Fill in sources
if source is None:
source_ids = df["source_id"].unique().tolist()
sources = session.scalars(select(source_class).filter(source_class.id.in_(source_ids))).all()
sources = session.scalars(
select(source_class).filter(source_class.id.in_(source_ids))
).all()
source_map = {source.id: source for source in sources}
df["source_id"] = df["source_id"].map(source_map)
df = df.rename(columns={"source_id": "source"})
Expand Down

0 comments on commit 6531e09

Please sign in to comment.