Merge pull request #3488 from vvorobjov/nest-server-mpi-serialization

Update the data serialization in hl_api_server.py

Author: terhorstd

pynest/nest/lib/hl_api_types.py

@@ -1286,6 +1286,8 @@ def serialize_data(data):
 
     if isinstance(data, (numpy.ndarray, NodeCollection)):
         return data.tolist()
+    if isinstance(data, (numpy.integer)):
+        return int(data)
     elif isinstance(data, SynapseCollection):
         # Get full information from SynapseCollection
         return serialize_data(data.get())

pynest/nest/server/hl_api_server.py

@@ -261,7 +261,7 @@ def do_call(call_name, args=[], kwargs={}):
         log(call_name, f"local call, args={args}, kwargs={kwargs}")
         master_response = call(*args, **kwargs)
 
-    response = [master_response]
+    response = [nest.serialize_data(master_response)]
     if mpi_comm is not None:
         log(call_name, "waiting for response gather")
         response = mpi_comm.gather(response[0], root=0)
@@ -277,7 +277,7 @@ def route_exec():
     if EXEC_CALL_ENABLED:
         args, kwargs = get_arguments(request)
         response = do_call("exec", args, kwargs)
-        return jsonify(response)
+        return jsonify(nest.serialize_data(response))
     else:
         flask.abort(
             403,
@@ -307,7 +307,7 @@ def route_api_call(call):
     args, kwargs = get_arguments(request)
     log("route_api_call", f"call={call}, args={args}, kwargs={kwargs}")
     response = api_client(call, args, kwargs)
-    return jsonify(response)
+    return jsonify(nest.serialize_data(response))
 
 
 # ----------------------