Fix docs for distiset.save_to_disk kwargs (#745)

argilla-io · Jun 21, 2024 · 473408a · 473408a
1 parent 3910aca
commit 473408a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/distilabel/distiset.py b/src/distilabel/distiset.py
@@ -270,7 +270,7 @@ def save_to_disk(
         Examples:
             ```python
             # Save your distiset in a local folder:
-            >>> distiset.save_to_disk(dataset_path="my-distiset")
+            >>> distiset.save_to_disk(distiset_path="my-distiset")
             # Save your distiset in a remote storage:
             >>> storage_options = {
             ...     "key": os.environ["S3_ACCESS_KEY"],
@@ -280,7 +280,7 @@ def save_to_disk(
             ...         "region_name": os.environ["S3_REGION"],
             ...     },
             ... }
-            >>> distiset.save_to_disk(dataset_path="my-distiset", storage_options=storage_options)
+            >>> distiset.save_to_disk(distiset_path="my-distiset", storage_options=storage_options)
             ```
         """
         distiset_path = str(distiset_path)