diff --git a/core/testcontainers/core/container.py b/core/testcontainers/core/container.py index f0882a4c..dcf576ea 100644 --- a/core/testcontainers/core/container.py +++ b/core/testcontainers/core/container.py @@ -169,7 +169,7 @@ def maybe_emulate_amd64(self) -> Self: return self.with_kwargs(platform="linux/amd64") return self - def waiting_for(self, strategy: WaitStrategy) -> "DockerContainer": + def waiting_for(self, strategy: WaitStrategy) -> Self: """Set a wait strategy to be used after container start.""" self._wait_strategy = strategy return self