From d3ef22ab173ac269ae8541a1d51e1028f736f2dd Mon Sep 17 00:00:00 2001 From: Sophie Tan Date: Fri, 6 Oct 2023 14:29:18 -0400 Subject: [PATCH] Clean up --- src/snowflake/snowpark/mock/plan.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/src/snowflake/snowpark/mock/plan.py b/src/snowflake/snowpark/mock/plan.py index 328c8cbbdab..34a97492d62 100644 --- a/src/snowflake/snowpark/mock/plan.py +++ b/src/snowflake/snowpark/mock/plan.py @@ -330,18 +330,6 @@ def execute_mock_plan( # Compute drop duplicates res_df = res_df.drop_duplicates() - elif operator == INTERSECT: - # Dedup all none rows - if res_df.isnull().all(axis=1).where(lambda x: x).count() > 1: - res_df = res_df.drop(index=res_df.isnull().all(axis=1).index[1:]) - - # Keep all none rows in res_df if there are all none rows in cur_df - keep_nones = cur_df.isnull().all(axis=1).any() - - res_df = res_df[ - (res_df.isin(cur_df.values.ravel()).all(axis=1)).values - | (keep_nones & res_df.isnull().all(axis=1).values) - ].drop_duplicates() else: raise NotImplementedError( f"[Local Testing] SetStatement operator {operator} is currently not implemented."