Skip to content

Commit a98c9e9

Browse files
committed
1 parent fce23a4 commit a98c9e9

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

core/testcontainers/core/waiting_utils.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,13 +15,15 @@
1515
import re
1616
import time
1717
import traceback
18-
from typing import Any, Callable, Iterable, Mapping, Optional
19-
18+
from typing import Any, Callable, Iterable, Mapping, Optional, TYPE_CHECKING
2019
import wrapt
2120

2221
from . import config
2322
from .utils import setup_logger
2423

24+
if TYPE_CHECKING:
25+
from .container import DockerContainer
26+
2527
logger = setup_logger(__name__)
2628

2729

@@ -74,8 +76,8 @@ def wait_for(condition: Callable[..., bool]) -> bool:
7476
return condition()
7577

7678

77-
def wait_for_logs(container, predicate: Callable, timeout: Optional[float] = None,
78-
interval: float = 1) -> float:
79+
def wait_for_logs(container: "DockerContainer", predicate: Callable,
80+
timeout: Optional[float] = None, interval: float = 1) -> float:
7981
"""
8082
Wait for the container to emit logs satisfying the predicate.
8183

0 commit comments

Comments
 (0)