瀏覽代碼

starpupy: use pickle.loads

HE Kun 4 年之前
父節點
當前提交
1c4e909d1d
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      starpupy/src/starpu_task_wrapper.c

+ 1 - 1
starpupy/src/starpu_task_wrapper.c

@@ -58,7 +58,7 @@ static char* starpu_cloudpickle_dumps(PyObject *obj, PyObject **obj_bytes, Py_ss
 
 static PyObject* starpu_cloudpickle_loads(char* pyString, Py_ssize_t pyString_size)
 {
-	PyObject *cloudpickle_module = PyImport_ImportModule("cloudpickle");
+	PyObject *cloudpickle_module = PyImport_ImportModule("pickle");
 	PyObject *loads = PyObject_GetAttrString(cloudpickle_module, "loads");
 	PyObject *obj_bytes_str = PyBytes_FromStringAndSize(pyString, pyString_size);
 	PyObject *obj = PyObject_CallFunctionObjArgs(loads, obj_bytes_str, NULL);