🎨 Add missing columns docstring

dyvenia · Jan 10, 2025 · d1a67bc · d1a67bc
1 parent 9cac055
commit d1a67bc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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