Skip to content

Commit

Permalink
Switch slurm worker to infer slurmrestd
Browse files Browse the repository at this point in the history
  • Loading branch information
rstyd committed Nov 26, 2024
1 parent 2f7dfce commit 8211781
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion beeflow/common/worker/slurm_worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
import requests

from beeflow.common import log as bee_logging
import beeflow.common.worker.utils as worker_utils
from beeflow.common.worker.worker import (Worker, WorkerError)
from beeflow.common import validation
from beeflow.common.worker.utils import get_state_sacct
Expand Down Expand Up @@ -154,8 +155,9 @@ def submit_task(self, task):
class SlurmrestdWorker(BaseSlurmWorker):
"""Worker class for when slurmrestd is available."""

def __init__(self, bee_workdir, openapi_version, **kwargs):
def __init__(self, bee_workdir, **kwargs):
"""Create a new Slurmrestd Worker object."""
openapi_version = worker_utils.get_slurmrestd_version()
super().__init__(bee_workdir=bee_workdir, **kwargs)
# Pull slurm socket configs from kwargs (Uses getpass.getuser() instead
# of os.getlogin() because of an issue with using getlogin() without a
Expand Down

0 comments on commit 8211781

Please sign in to comment.