Skip to content

Commit

Permalink
Add support for separate external address for SpecCluster scheduler (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
jacobtomlinson authored and mrocklin committed Aug 17, 2019
1 parent e16837c commit 31e7754
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion distributed/deploy/spec.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ class ProcessInterface:

def __init__(self):
self.address = None
self.external_address = None
self.lock = asyncio.Lock()
self.status = "created"

Expand Down Expand Up @@ -225,7 +226,7 @@ async def _start(self):
self.status = "starting"
self.scheduler = await self.scheduler
self.scheduler_comm = rpc(
self.scheduler.address,
getattr(self.scheduler, "external_address", None) or self.scheduler.address,
connection_args=self.security.get_connection_args("client"),
)
await super()._start()
Expand Down

0 comments on commit 31e7754

Please sign in to comment.