소스 검색

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);