Skip to content

Commit 91b78bf

Browse files
authored
Merge pull request #1992 from codalab/workerPackagesUpdate
Updated requests websockets and pyyaml of compute worker to fix some vulnerabilities
2 parents 1d6169b + fce7f1d commit 91b78bf

File tree

3 files changed

+419
-180
lines changed

3 files changed

+419
-180
lines changed

compute_worker/compute_worker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
import websockets
2222
import yaml
2323
from billiard.exceptions import SoftTimeLimitExceeded
24-
from celery import Celery, task, utils
24+
from celery import Celery, shared_task, utils
2525
from kombu import Queue, Exchange
2626
from urllib3 import Retry
2727

@@ -113,7 +113,7 @@ class ExecutionTimeLimitExceeded(Exception):
113113
# The main compute worker entrypoint, this is how a job is ran at the highest
114114
# level.
115115
# -----------------------------------------------------------------------------
116-
@task(name="compute_worker_run")
116+
@shared_task(name="compute_worker_run")
117117
def run_wrapper(run_args):
118118
logger.info(f"Received run arguments: {run_args}")
119119
run = Run(run_args)

0 commit comments

Comments
 (0)