WIP AI tests pass

Author: sfc-gh-jkew

src/snowflake/snowpark/modin/plugin/_internal/aggregation_utils.py

@@ -579,7 +579,7 @@ def _create_pandas_to_snowpark_pandas_aggregation_map(
             supported_in_pivot=True,
         ),
         "nunique": _SnowparkPandasAggregation(
-            axis_0_aggregation=lambda c: count(c.distinct()),
+            axis_0_aggregation=count_distinct,
             preserves_snowpark_pandas_types=False,
             supported_in_pivot=True,
         ),