fix formatting

Author: dorlib

instrumentation/opentelemetry-instrumentation-aiokafka/src/opentelemetry/instrumentation/aiokafka/utils.py

@@ -161,12 +161,18 @@ async def _extract_send_partition(
         key = _extract_send_key(args, kwargs)
         value = _extract_send_value(args, kwargs)
         partition = _extract_argument("partition", 3, None, args, kwargs)
-        key_bytes = instance._key_serializer(key) if instance._key_serializer else key  # type: ignore[reportUnknownMemberType]
-        value_bytes = instance._value_serializer(value) if instance._value_serializer else value  # type: ignore[reportUnknownMemberType]
+        key_bytes = (
+            instance._key_serializer(key) if instance._key_serializer else key
+        )  # type: ignore[reportUnknownMemberType]
+        value_bytes = (
+            instance._value_serializer(value)
+            if instance._value_serializer
+            else value
+        )  # type: ignore[reportUnknownMemberType]
         valid_types = (bytes, bytearray, memoryview, type(None))
         if (
-            type(key_bytes) not in valid_types
-            or type(value_bytes) not in valid_types
+            type(key_bytes) not in valid_types  # type: ignore[reportUnknownArgumentType]
+            or type(value_bytes) not in valid_types  # type: ignore[reportUnknownArgumentType]
         ):
             return None
 

instrumentation/opentelemetry-instrumentation-aiokafka/tests/test_utils.py

@@ -412,7 +412,9 @@ async def test_extract_send_partition_with_custom_serializers(self):
         args = [self.topic_name, test_value, test_key]
         kwargs = {"partition": 0, "headers": self.headers}
 
-        result = await _extract_send_partition(aiokafka_instance_mock, args, kwargs)
+        result = await _extract_send_partition(
+            aiokafka_instance_mock, args, kwargs
+        )
 
         self.assertEqual(result, 1)
         value_serializer.assert_called_once_with(test_value)