fix dataframe index when downloading cleanlab columns (#213)

* reset cl cols index * bump version (1.3.3)
cleanlab · Apr 9, 2024 · a4e1ae4 · a4e1ae4
1 parent 188662c
commit a4e1ae4
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 7 deletions.
diff --git a/cleanlab_studio/internal/api/api.py b/cleanlab_studio/internal/api/api.py
@@ -290,7 +290,7 @@ def download_cleanlab_columns(
     cleanset_json_io = io.StringIO(cleanset_json)
     cleanset_pd: pd.DataFrame = pd.read_json(cleanset_json_io, orient="table")
     cleanset_pd.rename(columns={"id": id_col}, inplace=True)
-    cleanset_pd.sort_values(by=id_col, inplace=True)
+    cleanset_pd.sort_values(by=id_col, inplace=True, ignore_index=True)
     return cleanset_pd
 
 

diff --git a/cleanlab_studio/studio/studio.py b/cleanlab_studio/studio/studio.py
@@ -138,11 +138,6 @@ def download_cleanlab_columns(
             include_project_details=include_project_details,
             to_spark=to_spark,
         )
-        if "cleanlab_row_ID" in rows_df.columns:
-            if to_spark:
-                rows_df.sort("cleanlab_row_ID")
-            else:
-                rows_df.sort_values(by="cleanlab_row_ID")
         return rows_df
 
     def apply_corrections(self, cleanset_id: str, dataset: Any, keep_excluded: bool = False) -> Any:

diff --git a/cleanlab_studio/version.py b/cleanlab_studio/version.py
@@ -1,4 +1,4 @@
-__version__ = "2.0.0"
+__version__ = "2.0.1"
 
 SCHEMA_VERSION = "0.2.0"
 MIN_SCHEMA_VERSION = "0.1.0"