Skip to content

Commit

Permalink
Wyd/dev conflict check
Browse files Browse the repository at this point in the history
  • Loading branch information
BIGWangYuDong authored and 周丰哲 committed Dec 14, 2023
1 parent 89e16c5 commit 42c7d51
Showing 1 changed file with 12 additions and 11 deletions.
23 changes: 12 additions & 11 deletions opencompass/runners/slurm.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 42c7d51

Please sign in to comment.