[typing] Add typing for user facing functions

neuralprophet/configure.py

@@ -398,7 +398,7 @@ def __post_init__(self):
 @dataclass
 class Regressor:
     reg_lambda: Optional[float]
-    normalize: str
+    normalize: Union[str, bool]
     mode: str
 
 
@@ -418,7 +418,7 @@ class Event:
 
 @dataclass
 class Holidays:
-    country: str
+    country: Union[str, List[str]]
     lower_window: int
     upper_window: int
     mode: str = "additive"

neuralprophet/df_utils.py

@@ -940,7 +940,14 @@ def split_considering_timestamp(df, n_lags, n_forecasts, inputs_overbleed, thres
     return df_train, df_val
 
 
-def split_df(df, n_lags, n_forecasts, valid_p=0.2, inputs_overbleed=True, local_split=False):
+def split_df(
+    df: pd.DataFrame,
+    n_lags: int,
+    n_forecasts: int,
+    valid_p: float = 0.2,
+    inputs_overbleed: bool = True,
+    local_split: bool = False,
+):
     """Splits timeseries df into train and validation sets.
 
     Prevents overbleed of targets. Overbleed of inputs can be configured.
@@ -1553,7 +1560,7 @@ def join_dfs_after_data_drop(predicted, df, merge=False):
         return df_merged.rename_axis("ds").reset_index()
 
 
-def add_quarter_condition(df):
+def add_quarter_condition(df: pd.DataFrame):
     """Adds columns for conditional seasonalities to the df.
 
     Parameters
@@ -1578,7 +1585,7 @@ def add_quarter_condition(df):
     return df
 
 
-def add_weekday_condition(df):
+def add_weekday_condition(df: pd.DataFrame):
     """Adds columns for conditional seasonalities to the df.
 
     Parameters