Skip to content

Commit 2dfb01f

Browse files
committed
tests: fix Generator type annotations
1 parent 024a1d7 commit 2dfb01f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/integration/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ async def fixture_enapter_mqtt_client(
3030
@pytest.fixture(scope="session", name="mosquitto_container")
3131
def fixture_mosquitto_container(
3232
docker_client: docker.DockerClient,
33-
) -> Generator[docker.models.containers.Container, None]:
33+
) -> Generator[docker.models.containers.Container, None, None]:
3434
name = "enapter-python-sdk-integration-tests-mosquitto"
3535

3636
try:
@@ -64,7 +64,7 @@ def random_unused_port() -> int:
6464

6565

6666
@pytest.fixture(name="docker_client", scope="session")
67-
def fixture_docker_client() -> Generator[docker.DockerClient, None]:
67+
def fixture_docker_client() -> Generator[docker.DockerClient, None, None]:
6868
docker_client = docker.from_env()
6969
try:
7070
yield docker_client

tests/unit/test_log.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def test_stack_info(self) -> None:
6161

6262
@staticmethod
6363
@contextlib.contextmanager
64-
def configure(*args, **kwargs) -> Generator[None, None]:
64+
def configure(*args, **kwargs) -> Generator[None, None, None]:
6565
enapter.log.configure(*args, **kwargs)
6666
try:
6767
yield

0 commit comments

Comments
 (0)