# define packages that your function requires to run session.add_packages("scikit-learn==1.0.2", "pandas", "numpy") # UDF definition @udf(name='predict_churn',is_permanent = True, stage_location = '@MODELSTAGE', replace=True, session=session) def predict_churn(args: list) -> float: row = pd.DataFrame([args], columns=feature_columns) return model.predict(row)