diff --git a/src/viadot/sources/mediatool.py b/src/viadot/sources/mediatool.py index 73eea1dca..5a129f988 100644 --- a/src/viadot/sources/mediatool.py +++ b/src/viadot/sources/mediatool.py @@ -273,6 +273,9 @@ def fetch_and_transform( organization_id (str, optional): Organization ID. Defaults to None. vehicle_ids (list[str]): List of organization IDs. Defaults to None. media_type_ids (list[str]): List of media type IDs. Defaults to None. + columns (list[str], optional): If provided, a list of column names to + include in the DataFrame.By default, all columns will be included. + Defaults to None. if_empty (str, optional): What to do if a fetch produce no data. Defaults to "warn add_endpoint_suffix (bool, optional): If True, appends the endpoint name