Update sort_first_koalas.py

eds_scikit/utils/sort_first_koalas.py

@@ -4,24 +4,26 @@
 
 
 def sort_values_first_koalas(
-    dataframe: DataFrame, by_cols: List[str], cols: List[str], ascending: bool = True
+    dataframe: DataFrame, by_cols: List[str], cols: List[str], disambiguate_col: str, ascending: bool = True
 ) -> DataFrame:
-    """Use this function to obtain in koalas the same ouput as dataframe.sort_values(cols).groupby(by_cols).first() in pandas.
+    """Use this function to obtain in koalas the same ouput as dataframe.sort_values([*cols, disambiguate_col]).groupby(by_cols).first() in pandas.
 
-    If you want the output to be deterministic, provide an id column of your dataframe as the last element of variable cols.
+    disambiguate_col must be provided to make sure the output is deterministic 
 
     Parameters
     ----------
     dataframe : DataFrame
     by_cols : List[str]
     cols : List[str]
+    disambiguate_col : List[str]
     ascending : bool, optional
 
     Returns
     -------
     DataFrame
 
     """
+    cols = [*cols, disambiguate_col]
     for col in cols:
         dataframe_min_max = dataframe.groupby(by_cols, as_index=False)[col]
         dataframe_min_max = (