Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions dask_cloudprovider/gcp/instances.py
Original file line number Diff line number Diff line change
Expand Up @@ -662,17 +662,18 @@ def __init__(
self.scheduler_machine_type = machine_type
self.worker_machine_type = machine_type

self.ngpus = ngpus or self.config.get("ngpus")
ngpus_value = ngpus or self.config.get("ngpus")
self.ngpus = int(ngpus_value) if ngpus_value else None
if not self.ngpus:
self.scheduler_ngpus = (
scheduler_ngpus
if scheduler_ngpus is not None
else self.config.get("scheduler_ngpus", 0)
else int(self.config.get("scheduler_ngpus") or 0)
)
self.worker_ngpus = (
worker_ngpus
if worker_ngpus is not None
else self.config.get("worker_ngpus", 0)
else int(self.config.get("worker_ngpus") or 0)
)
else:
if scheduler_ngpus is not None or worker_ngpus is not None:
Expand Down
Loading