Fixed read from joblib 3

exasol · Oct 8, 2024 · b2278a9 · b2278a9
1 parent a713942
commit b2278a9
Showing 1 changed file with 2 additions and 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
@@ -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