Unverified Commit 701ecbb2 authored by binary-husky's avatar binary-husky Committed by GitHub
Browse files

[Fix] python path bug (#1063)



* fix relative path bug

* format

---------
Co-authored-by: default avatarhmp <505030475@qq.com>
Co-authored-by: default avatarLeymore <zfz-960727@163.com>
parent 048d41a1
......@@ -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)
......
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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment