Fix restore_savestate

neuronsimulator · Nov 20, 2024 · 649d4a4 · 649d4a4
1 parent 28ebfcc
commit 649d4a4
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/nrnpython/nrnpy_hoc.cpp b/src/nrnpython/nrnpy_hoc.cpp
@@ -2622,15 +2622,11 @@ static void nrnpy_store_savestate_(char** save_data, uint64_t* save_data_size) {
 
 static void nrnpy_restore_savestate_(int64_t size, char* data) {
     if (restore_savestate_) {
-        PyObject* args = PyTuple_New(1);
-        auto py_data = nb::steal(PyByteArray_FromStringAndSize(data, size));
+        nb::bytearray py_data(data, size);
         if (!py_data) {
             hoc_execerror("SaveState:", "Data restore failure.");
         }
-        // note: PyTuple_SetItem steals a ref to py_data
-        PyTuple_SetItem(args, 0, py_data.release().ptr());
-        auto result = nb::steal(PyObject_CallObject(restore_savestate_, args));
-        Py_DECREF(args);
+        auto result = nb::steal(PyObject_CallOneArg(restore_savestate_, py_data.ptr()));
         if (!result) {
             hoc_execerror("SaveState:", "Data restore failure.");
         }