Skip to content

Commit b392ca6

Browse files
committed
chore: clarify Network init signature
1 parent cac19d3 commit b392ca6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

core/testcontainers/core/network.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,10 @@ class Network(object):
2121
Network context manager for programmatically connecting containers.
2222
"""
2323

24-
def __init__(self, docker_client_kw: Optional[dict] = None, **kwargs) -> None:
24+
def __init__(self, docker_client_kw: Optional[dict] = None, docker_network_kw: Optional[dict] = None) -> None:
2525
self.name = str(uuid.uuid4())
2626
self._docker = DockerClient(**(docker_client_kw or {}))
27-
self._kwargs = kwargs
27+
self._docker_network_kw = docker_network_kw or {}
2828

2929
def connect(self, container_id: str, network_aliases: Optional[list] = None):
3030
self._network.connect(container_id, aliases=network_aliases)
@@ -33,7 +33,7 @@ def remove(self) -> None:
3333
self._network.remove()
3434

3535
def __enter__(self) -> 'Network':
36-
self._network = self._docker.client.networks.create(self.name, **self._kwargs)
36+
self._network = self._docker.client.networks.create(self.name, **self._docker_network_kw)
3737
self.id = self._network.id
3838
return self
3939

0 commit comments

Comments
 (0)