jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Fix] python path bug (#1063)

Browse Source 
* fix relative path bug

* format

---------

Co-authored-by: hmp <505030475@qq.com>
Co-authored-by: Leymore <zfz-960727@163.com>
This commit is contained in:
binary-husky 2024-04-26 21:58:45 +08:00 committed by GitHub
parent 048d41a1c4
commit 701ecbb292
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
opencompass/tasks/openicl_eval.py
View File

@ -2,13 +2,14 @@ import argparse
import copy
import fnmatch
import math
import os
import os.path as osp
import re
import statistics
import sys
import time
from collections import Counter
from inspect import signature
from shutil import which
from typing import List, Optional
import mmengine
@ -76,8 +77,9 @@ class OpenICLEvalTask(BaseTask):
'task', {}).get('dump_details', False)
def get_command(self, cfg_path, template):
sys.path.append(os.getcwd())
script_path = __file__
python = 'python3' if which('python3') else 'python'
python = sys.executable
command = f'{python} {script_path} {cfg_path}'
return template.format(task_cmd=command)

6
opencompass/tasks/openicl_infer.py
View File

@ -1,8 +1,9 @@
import argparse
import os
import os.path as osp
import random
import sys
import time
from shutil import which
from typing import Any
from mmengine.config import Config, ConfigDict
@ -42,6 +43,7 @@ class OpenICLInferTask(BaseTask):
template (str): The template which have '{task_cmd}' to format
the command.
"""
sys.path.append(os.getcwd())
script_path = __file__
backend_keys = ['VLLM', 'Lmdeploy']
use_backend = any(
@ -54,7 +56,7 @@ class OpenICLInferTask(BaseTask):
f'--nproc_per_node {self.num_procs} '
f'{script_path} {cfg_path}')
else:
python = 'python3' if which('python3') else 'python'
python = sys.executable
command = f'{python} {script_path} {cfg_path}'
return template.format(task_cmd=command)