diff --git a/google/api_core/_rest_streaming_base.py b/google/api_core/_rest_streaming_base.py index 406184d4..1a870985 100644 --- a/google/api_core/_rest_streaming_base.py +++ b/google/api_core/_rest_streaming_base.py @@ -97,7 +97,9 @@ def _process_chunk(self, chunk: str): def _grab(self): # Add extra quotes to make json.loads happy. if issubclass(self._response_message_cls, proto.Message): - return self._response_message_cls.from_json(self._ready_objs.popleft(), ignore_unknown_fields=True) + return self._response_message_cls.from_json( + self._ready_objs.popleft(), ignore_unknown_fields=True + ) elif issubclass(self._response_message_cls, google.protobuf.message.Message): return Parse(self._ready_objs.popleft(), self._response_message_cls()) else: diff --git a/tests/unit/test_rest_streaming.py b/tests/unit/test_rest_streaming.py index 006adef7..9ab90d7b 100644 --- a/tests/unit/test_rest_streaming.py +++ b/tests/unit/test_rest_streaming.py @@ -78,7 +78,6 @@ def iter_content(self, *args, **kwargs): ) - @pytest.mark.parametrize( "random_split,resp_message_is_proto_plus", [(False, True), (False, False)],