minor PR feedback

ml6team · May 4, 2023 · 924a59d · 924a59d
1 parent 834ed49
commit 924a59d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fondant/dataset.py b/fondant/dataset.py
@@ -127,7 +127,7 @@ def write_index(self, df: dd.DataFrame):
         """
         Write the index dataframe to a remote location.
         Args:
-            df: The input Dask dataframe.
+            df: The output Dask dataframe returned by the user.
         """
         remote_path = self.manifest.index.location
         index_columns = list(self.manifest.index.fields.keys())
@@ -147,8 +147,8 @@ def write_subsets(self, df: dd.DataFrame, spec: FondantComponentSpec):
         Write all subsets of the Dask dataframe to a remote location.
 
         Args:
-            df (dask.dataframe.DataFrame): The input Dask dataframe.
-            spec (FondantComponentSpec): The specification of the output subsets.
+            df (dask.dataframe.DataFrame): The output Dask dataframe returned by the user.
+            spec (FondantComponentSpec): The component specification.
 
         Raises:
             ValueError: If a field defined in an output subset is not present in the user