from snowflake.snowpark import functions as F new_sample = session.table('TELCO_DATASET').sample(n = 500) new_sample.select( col('CUSTOMERID'), col('CHURNVALUE'), F.call_udf('predict_churn', F.array_construct(*feature_columns)) .alias('PREDICTED_CHURN')) .write.mode('overwrite') .saveAsTable('churn_detection')