merge main

aditya1503 · aditya1503 · commit 692efe4d880f · 2023-12-13T22:19:45.000+05:30
diff --git a/cleanlab_studio/studio/studio.py b/cleanlab_studio/studio/studio.py
@@ -144,10 +144,10 @@ def apply_corrections(self, cleanset_id: str, dataset: Any, keep_excluded: bool
             cl_cols = self.download_cleanlab_columns(
                 cleanset_id, to_spark=False, include_project_details=True
             )
-            snowflake_corrected_ds: snowpark.DataFrame = apply_corrections_snowpark_df(
+            corrected_ds: snowpark.DataFrame = apply_corrections_snowpark_df(
                 dataset, cl_cols, id_col, label_col, keep_excluded
             )
-            return snowflake_corrected_ds
+            return corrected_ds
 
         elif _pyspark_exists and isinstance(dataset, pyspark.sql.DataFrame):
             cl_cols = self.download_cleanlab_columns(

Original file line number	Diff line number	Diff line change
`@@ -144,10 +144,10 @@ def apply_corrections(self, cleanset_id: str, dataset: Any, keep_excluded: bool`
`144`	`144`	`cl_cols = self.download_cleanlab_columns(`
`145`	`145`	`cleanset_id, to_spark=False, include_project_details=True`
`146`	`146`	`)`
`147`		`- snowflake_corrected_ds: snowpark.DataFrame = apply_corrections_snowpark_df(`
	`147`	`+ corrected_ds: snowpark.DataFrame = apply_corrections_snowpark_df(`
`148`	`148`	`dataset, cl_cols, id_col, label_col, keep_excluded`
`149`	`149`	`)`
`150`		`- return snowflake_corrected_ds`
	`150`	`+ return corrected_ds`
`151`	`151`
`152`	`152`	`elif _pyspark_exists and isinstance(dataset, pyspark.sql.DataFrame):`
`153`	`153`	`cl_cols = self.download_cleanlab_columns(`