Update serialization.jl (#494)

making it possible to use another package instead of `pickle` for serialization (e.g. `dill`)
JuliaPy · May 12, 2024 · f898a2c · f898a2c
1 parent 662e7ae
commit f898a2c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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