From b86ca29fde751fdb489c1ae581df245cededc4f9 Mon Sep 17 00:00:00 2001 From: Ruben Fiszel Date: Tue, 4 Apr 2023 14:25:27 +0200 Subject: [PATCH] fix polar handling --- backend/windmill-api/src/jobs.rs | 1 - backend/windmill-worker/src/worker.rs | 8 ++++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/backend/windmill-api/src/jobs.rs b/backend/windmill-api/src/jobs.rs index 8a7380b56a..9f3df5b21a 100644 --- a/backend/windmill-api/src/jobs.rs +++ b/backend/windmill-api/src/jobs.rs @@ -1398,7 +1398,6 @@ pub async fn run_wait_result_job_by_path_get( Path((w_id, script_path)): Path<(String, StripPath)>, Query(run_query): Query, ) -> error::JsonResult { - tracing::error!("run_wait_result_job_by_path_get {:#?}", run_query.payload); let payload_r = run_query .payload .map(decode_payload) diff --git a/backend/windmill-worker/src/worker.rs b/backend/windmill-worker/src/worker.rs index b0994d03c1..0b1c30122f 100644 --- a/backend/windmill-worker/src/worker.rs +++ b/backend/windmill-worker/src/worker.rs @@ -1918,8 +1918,12 @@ for k, v in list(args.items()): try: res = inner_script.main(**args) - if type(res).__name__ == 'DataFrame': - res = res.values.tolist() + typ = type(res) + if typ.__name__ == 'DataFrame': + if typ.__module__ == 'pandas.core.frame': + res = res.values.tolist() + elif typ.__module__ == 'polars.dataframe.frame': + res = res.rows() res_json = json.dumps(res, separators=(',', ':'), default=str).replace('\n', '') with open("result.json", 'w') as f: f.write(res_json)