From 42c7d5174fe425558d52301cde235639f11230c0 Mon Sep 17 00:00:00 2001 From: wangyudong Date: Thu, 14 Dec 2023 08:08:44 +0000 Subject: [PATCH] Wyd/dev conflict check --- opencompass/runners/slurm.py | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/opencompass/runners/slurm.py b/opencompass/runners/slurm.py index 74b2e6dcf..aad8c7c04 100644 --- a/opencompass/runners/slurm.py +++ b/opencompass/runners/slurm.py @@ -35,17 +35,18 @@ class SlurmRunner(BaseRunner): For example ['-c 12', '-w node1']. Defaults to None. """ - def __init__(self, - task: ConfigDict, - task_prefix: str = '', # INTERNAL - max_num_workers: int = 32, - retry: int = 2, - partition: str = None, - quotatype: str = None, - qos: str = None, - debug: bool = False, - lark_bot_url: str = None, - extra_command: Optional[List[str]] = None): + def __init__( + self, + task: ConfigDict, + task_prefix: str = '', # INTERNAL + max_num_workers: int = 32, + retry: int = 2, + partition: str = None, + quotatype: str = None, + qos: str = None, + debug: bool = False, + lark_bot_url: str = None, + extra_command: Optional[List[str]] = None): super().__init__(task=task, debug=debug, lark_bot_url=lark_bot_url) self.max_num_workers = max_num_workers self.retry = retry