refactor: move type annotations

Signed-off-by: F.N. Claessen <felix@seita.nl>
SeitaBV · May 13, 2024 · 7f4b8d6 · 7f4b8d6
1 parent 18829c5
commit 7f4b8d6
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/timely_beliefs/beliefs/classes.py b/timely_beliefs/beliefs/classes.py
@@ -40,6 +40,7 @@
 from timely_beliefs.beliefs import probabilistic_utils
 from timely_beliefs.beliefs import utils as belief_utils
 from timely_beliefs.beliefs.utils import is_pandas_structure, is_tb_structure, meta_repr
+from timely_beliefs.beliefs.time_utils import DatetimeLike, TimedeltaLike
 from timely_beliefs.db_base import Base
 from timely_beliefs.sensors import utils as sensor_utils
 from timely_beliefs.sensors.classes import DBSensor, Sensor, SensorDBMixin
@@ -48,8 +49,6 @@
 from timely_beliefs.sources.classes import BeliefSource, DBBeliefSource
 
 METADATA = ["sensor", "event_resolution"]
-DatetimeLike = Union[datetime, str, pd.Timestamp]
-TimedeltaLike = Union[timedelta, str, pd.Timedelta]
 JoinTarget = Union[
     Selectable,
     type,

diff --git a/timely_beliefs/beliefs/time_utils.py b/timely_beliefs/beliefs/time_utils.py
@@ -1,10 +1,14 @@
 from __future__ import annotations
 
-from datetime import timedelta
+from datetime import datetime, timedelta
+from typing import Union
 
 from isodate import parse_duration
 import pandas as pd
 
+DatetimeLike = Union[datetime, str, pd.Timestamp]
+TimedeltaLike = Union[timedelta, str, pd.Timedelta]
+
 
 def to_max_timedelta(duration: str | pd.DateOffset | timedelta) -> pd.Timedelta:
     """Determine the maximum pd.Timedelta for a given ISO duration string or Pandas DateOffset object.

diff --git a/timely_beliefs/beliefs/utils.py b/timely_beliefs/beliefs/utils.py
@@ -3,7 +3,6 @@
 import math
 import warnings
 from datetime import datetime, timedelta
-from typing import Union
 
 import numpy as np
 import pandas as pd
@@ -21,11 +20,9 @@
     get_median_belief,
     probabilistic_nan_mean,
 )
-from timely_beliefs.beliefs.time_utils import iso_duration_to_offset, to_max_timedelta
+from timely_beliefs.beliefs.time_utils import iso_duration_to_offset, to_max_timedelta, TimedeltaLike
 from timely_beliefs.sources import utils as source_utils
 
-TimedeltaLike = Union[timedelta, str, pd.Timedelta]
-
 
 def select_most_recent_belief(
     df: "classes.BeliefsDataFrame",