From e104c010aec5aab37022d8a7ab8269841110d997 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicolas=20H=C3=B6ning?= Date: Fri, 1 Mar 2024 10:41:09 +0100 Subject: [PATCH] black with version we use in pre-commit MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Nicolas Höning --- timely_beliefs/beliefs/classes.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/timely_beliefs/beliefs/classes.py b/timely_beliefs/beliefs/classes.py index fc7b1462..ee2bd864 100644 --- a/timely_beliefs/beliefs/classes.py +++ b/timely_beliefs/beliefs/classes.py @@ -1173,12 +1173,16 @@ def _for_each_belief( index_names.extend( ["event_start"] if "event_start" in df.index.names - else ["event_end"] if "event_end" in df.index.names else [] + else ["event_end"] + if "event_end" in df.index.names + else [] ) index_names.extend( ["belief_time"] if "belief_time" in df.index.names - else ["belief_horizon"] if "belief_horizon" in df.index.names else [] + else ["belief_horizon"] + if "belief_horizon" in df.index.names + else [] ) if collective_beliefs is False: index_names.append("source") @@ -1556,7 +1560,8 @@ def form_beliefs( belief_time: datetime, source: BeliefSource, event_start: datetime = None, - event_time_window: tuple[datetime, datetime] | None = ( + event_time_window: tuple[datetime, datetime] + | None = ( None, None, ), @@ -2117,7 +2122,9 @@ def assign_sensor_and_event_resolution(df, sensor, event_resolution): df.event_resolution = ( event_resolution if event_resolution - else sensor.event_resolution if sensor else None + else sensor.event_resolution + if sensor + else None )