Skip to content

Commit cbfbbf5

Browse files
committed
removing occurences of dask-gateway
1 parent 7320d64 commit cbfbbf5

File tree

11 files changed

+38
-616
lines changed

11 files changed

+38
-616
lines changed

packages/pytest-simcore/src/pytest_simcore/dask_gateway.py

Lines changed: 0 additions & 119 deletions
This file was deleted.

services/director-v2/src/simcore_service_director_v2/api/routes/clusters.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -200,9 +200,7 @@ async def test_cluster_connection(
200200
cluster_auth: ClusterPing,
201201
):
202202
try:
203-
return await test_scheduler_endpoint(
204-
endpoint=cluster_auth.endpoint, authentication=cluster_auth.authentication
205-
)
203+
return await test_scheduler_endpoint(endpoint=cluster_auth.endpoint)
206204

207205
except ConfigurationError as e:
208206
raise HTTPException(
@@ -220,9 +218,7 @@ async def test_default_cluster_connection(
220218
settings: ComputationalBackendSettings = Depends(get_scheduler_settings),
221219
):
222220
cluster = settings.default_cluster
223-
return await test_scheduler_endpoint(
224-
endpoint=cluster.endpoint, authentication=cluster.authentication
225-
)
221+
return await test_scheduler_endpoint(endpoint=cluster.endpoint)
226222

227223

228224
@router.post(
@@ -237,6 +233,4 @@ async def test_specific_cluster_connection(
237233
clusters_repo: ClustersRepository = Depends(get_repository(ClustersRepository)),
238234
):
239235
cluster = await clusters_repo.get_cluster(user_id, cluster_id)
240-
return await test_scheduler_endpoint(
241-
endpoint=cluster.endpoint, authentication=cluster.authentication
242-
)
236+
return await test_scheduler_endpoint(endpoint=cluster.endpoint)

services/director-v2/src/simcore_service_director_v2/modules/dask_client.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -331,7 +331,6 @@ async def send_computation_tasks(
331331
)
332332
dask_utils.check_communication_with_scheduler_is_open(self.backend.client)
333333
dask_utils.check_scheduler_status(self.backend.client)
334-
await dask_utils.check_maximize_workers(self.backend.gateway_cluster)
335334
# NOTE: in case it's a gateway or it is an on-demand cluster
336335
# we do not check a priori if the task
337336
# is runnable because we CAN'T. A cluster might auto-scale, the worker(s)

services/director-v2/src/simcore_service_director_v2/utils/dask.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
from typing import Any, Final, NoReturn, ParamSpec, TypeVar, cast
66
from uuid import uuid4
77

8-
import dask_gateway # type: ignore[import-untyped]
98
import distributed
109
from aiopg.sa.engine import Engine
1110
from common_library.json_serialization import json_dumps
@@ -515,14 +514,6 @@ def check_scheduler_status(client: distributed.Client):
515514
raise ComputationalBackendNotConnectedError
516515

517516

518-
_LARGE_NUMBER_OF_WORKERS: Final[int] = 10000
519-
520-
521-
async def check_maximize_workers(cluster: dask_gateway.GatewayCluster | None) -> None:
522-
if cluster:
523-
await cluster.scale(_LARGE_NUMBER_OF_WORKERS)
524-
525-
526517
def _can_task_run_on_worker(
527518
task_resources: dict[str, Any], worker_resources: dict[str, Any]
528519
) -> bool:

0 commit comments

Comments
 (0)