From b2278a9c36868a76f857f545cc661d6cdff60133 Mon Sep 17 00:00:00 2001 From: ckunki Date: Tue, 8 Oct 2024 11:14:18 +0200 Subject: [PATCH] Fixed read from joblib 3 --- .../udf_framework/query_handler_runner_udf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/exasol_advanced_analytics_framework/udf_framework/query_handler_runner_udf.py b/exasol_advanced_analytics_framework/udf_framework/query_handler_runner_udf.py index 479475c..2f80523 100644 --- a/exasol_advanced_analytics_framework/udf_framework/query_handler_runner_udf.py +++ b/exasol_advanced_analytics_framework/udf_framework/query_handler_runner_udf.py @@ -41,14 +41,14 @@ def create_bucketfs_location_from_conn_object(bfs_conn_obj) -> bfs.path.PathLike def upload_via_joblib(location: bfs.path.PathLike, object: Any): buffer = BytesIO() joblib.dump(object, buffer) - location.write(buffer) + location.write(buffer.getvalue()) def read_via_joblib(location: bfs.path.PathLike) -> Any: buffer = BytesIO() for chunk in location.read(): buffer.write(chunk) - return joblib.load(buffer.getvalue()) + return joblib.load(buffer) @dataclasses.dataclass