Fix up import of cloudpickle to use vendored version

numba · Apr 26, 2021 · 58c4e32 · 58c4e32
1 parent b9a7fc1
commit 58c4e32
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/numba/core/serialize.py b/numba/core/serialize.py
@@ -7,9 +7,7 @@
 
 
 import pickle
-from cloudpickle import cloudpickle_fast
-
-cloudpickle = cloudpickle_fast
+from numba.cloudpickle import cloudpickle_fast as cloudpickle
 
 
 #
@@ -161,7 +159,7 @@ def is_serialiable(obj):
             return True
 
 
-class NumbaPickler(cloudpickle_fast.CloudPickler):
+class NumbaPickler(cloudpickle.CloudPickler):
     pass