Make format

Author: RaoulSchaffranek

pyk/src/pyk/rpc/rpc.py

@@ -3,18 +3,20 @@
 import json
 import logging
 from abc import ABC, abstractmethod
+from collections.abc import Iterator
 from dataclasses import dataclass
 from functools import partial
 from http.server import BaseHTTPRequestHandler, HTTPServer
 from typing import TYPE_CHECKING, NamedTuple
+
 from typing_extensions import Protocol
 
 from ..cli.cli import Options
 
 if TYPE_CHECKING:
     from collections.abc import Callable
     from pathlib import Path
-    from typing import Any, Final, Iterator
+    from typing import Any, Final
 
 
 _LOGGER: Final = logging.getLogger(__name__)
@@ -86,8 +88,7 @@ class JsonRpcBatchRequest(NamedTuple):
 class JsonRpcResult(ABC):
 
     @abstractmethod
-    def encode(self) -> Iterator[bytes]:
-        ...
+    def encode(self) -> Iterator[bytes]: ...
 
 
 @dataclass(frozen=True)
@@ -138,8 +139,7 @@ def encode(self) -> Iterator[bytes]:
                 yield b','
             else:
                 first = False
-            for chunk in result.encode():
-                yield chunk
+            yield from result.encode()
         yield b']'
 
 

pyk/src/tests/integration/test_json_rpc.py

@@ -1,11 +1,10 @@
 from __future__ import annotations
 
-from collections.abc import Iterator
 import json
 from http.client import HTTPConnection
 from threading import Thread
 from time import sleep
-from typing import TYPE_CHECKING, Iterator
+from typing import TYPE_CHECKING
 
 from pyk.cterm import CTerm
 from pyk.kast.inner import KApply, KSequence, KSort, KToken
@@ -16,6 +15,7 @@
 from pyk.testing import KRunTest
 
 if TYPE_CHECKING:
+    from collections.abc import Iterator
     from typing import Any
 
 
@@ -171,7 +171,7 @@ def exec_set_y(self, n: int) -> None:
 
     def exec_add(self) -> int:
         return self.x + self.y
-    
+
     def exec_streaming(self) -> Iterator[bytes]:
         yield b'{'
         yield b'"foo": "bar"'