fix lint

instrumentation/opentelemetry-instrumentation-confluent-kafka/tests/test_instrumentation.py

@@ -277,4 +277,4 @@ def test_producer_flush(self) -> None:
         producer = instrumentation.instrument_producer(producer)
         producer.produce(topic="topic-1", key="key-1", value="value-1")
         msg = producer.flush()
-        self.assertIsNotNone(msg)
+        self.assertIsNotNone(msg)

instrumentation/opentelemetry-instrumentation-confluent-kafka/tests/utils.py

@@ -1,6 +1,8 @@
-from confluent_kafka import Consumer, Producer
 from typing import Optional
 
+from confluent_kafka import Consumer, Producer
+
+
 class MockConsumer(Consumer):
     def __init__(self, queue, config):
         self._queue = queue
@@ -21,14 +23,14 @@ def poll(self, timeout=None):
 
 class MockedMessage:
     def __init__(
-            self,
-            topic: str,
-            partition: int,
-            offset: int,
-            headers,
-            key: Optional[str]=None,
-            value=Optional[str]=None
-        ):
+        self,
+        topic: str,
+        partition: int,
+        offset: int,
+        headers,
+        key: Optional[str] = None,
+        value: Optional[str] = None,
+    ):
         self._topic = topic
         self._partition = partition
         self._offset = offset
@@ -70,12 +72,12 @@ def produce(
                 offset=0,
                 headers=[],
                 key=kwargs.get("key"),
-                value=kwargs.get("value")
+                value=kwargs.get("value"),
             )
         )
 
-    def poll(self, timeout=None):
+    def poll(self, *args, **kwargs):
         return len(self._queue)
 
-    def flush(self, timeout=None):
+    def flush(self, *args, **kwargs):
         return len(self._queue)