From d181d6671d90ca7c0e350b61efa40e1b9b45e471 Mon Sep 17 00:00:00 2001 From: Stephan Sahm Date: Wed, 24 Apr 2024 16:52:29 +0200 Subject: [PATCH] Update serialization.jl making it possible to use another package instead of `pickle` for serialization (e.g. `dill`) --- src/Compat/serialization.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/Compat/serialization.jl b/src/Compat/serialization.jl index 5f600673..d5d65421 100644 --- a/src/Compat/serialization.jl +++ b/src/Compat/serialization.jl @@ -6,7 +6,7 @@ function serialize_py(s, x::Py) if pyisnull(x) serialize(s, nothing) else - b = pyimport("pickle").dumps(x) + b = pyimport(get(ENV, "JULIA_PYTHONCALL_PICKLE", "pickle")).dumps(x) serialize(s, pybytes_asvector(b)) end end @@ -16,7 +16,7 @@ function deserialize_py(s) if v === nothing pynew() else - pyimport("pickle").loads(pybytes(v)) + pyimport(get(ENV, "JULIA_PYTHONCALL_PICKLE", "pickle")).loads(pybytes(v)) end end