lint

Signed-off-by: Sidhant Kohli <sidhant.kohli@gmail.com>

Author: kohlisid

pynumaflow/mapper/__init__.py

@@ -14,5 +14,5 @@
     "MapServer",
     "MapAsyncServer",
     "MapMultiprocServer",
-    "AsyncMapMultiprocServer"
+    "AsyncMapMultiprocServer",
 ]

pynumaflow/mapper/_servicer/_async_servicer.py

@@ -89,6 +89,7 @@ async def _process_inputs(self, request_iterator, result_queue):
             _LOGGER.critical("MapFn Error in _process_inputs", exc_info=True)
         finally:
             await result_queue.put(STREAM_EOF)
+
     # async def _process_inputs(
     #     self,
     #     request_iterator: AsyncIterable[map_pb2.MapRequest],

tests/map/test_async_multiproc.py

@@ -14,9 +14,12 @@
 
 
 async def async_handler(keys, datum: Datum) -> Messages:
-    msg = f"payload:{datum.value.decode()} event_time:{datum.event_time} watermark:{datum.watermark}"
+    msg = (
+        f"payload:{datum.value.decode()} event_time:{datum.event_time} watermark:{datum.watermark}"
+    )
     return Messages(Message(value=msg.encode(), keys=keys))
 
+
 class TestAsyncMapMultiprocServer(unittest.TestCase):
     def setUp(self):
         self.base_sock_path = sock_prefix
@@ -25,7 +28,7 @@ def setUp(self):
             server_count=2,
             sock_path=self.base_sock_path,
             use_tcp=False,
-            server_info_file="/tmp/server_info"
+            server_info_file="/tmp/server_info",
         )
         self.process = Process(target=self.server.start)
         self.process.start()
@@ -80,4 +83,6 @@ def test_map_fn(self):
 
     def test_server_start(self):
         for path in self.socket_paths:
-            self.assertTrue(os.path.exists(path), f"Server socket {path} was not created successfully")
+            self.assertTrue(
+                os.path.exists(path), f"Server socket {path} was not created successfully"
+            )