Properly error in multiprocessing for sockets

materialsproject · Sep 22, 2022 · c49e901 · c49e901
1 parent 5f3d85d
commit c49e901
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/maggma/cli/multiprocessing.py b/src/maggma/cli/multiprocessing.py
@@ -199,9 +199,10 @@ async def multi(builder, num_processes, no_bars=False, socket=None):
     )
 
     if socket:
-        await socket.send("PING")
+        await socket.send_string("PING")
         message = await socket.recv()
         if message.decode("utf-8") != "PONG":
+            socket.close()
             raise RuntimeError("Stopping work as main node is not responding.")
 
     back_pressure_relief = back_pressured_get.release(processed_items)