Skip to content

Commit 44fafa1

Browse files
committed
Skip language build, use pre-built get_language
1 parent f70f5eb commit 44fafa1

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/codetext/utils/utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -96,17 +96,17 @@ def parse_code(raw_code: str, language: str='Auto', tree_sitter_path: str=None)
9696
parser = Parser()
9797
try:
9898
from tree_sitter_languages import get_language, get_parser
99-
parser = get_parser(get_language(language))
99+
language = get_language(language)
100100
except ImportError:
101101
# Work-around when pre-built binaries wheels for tree-sitter-languages are not available
102102
logger.warning(f"Troubled importing 'tree-sitter-languages', attemp to look for pre-built binaries in the workspace")
103103
ts_lang_path = os.path.join(load_path, 'tree-sitter', f'{language}.so')
104104
if not os.path.exists(ts_lang_path):
105105
logger.warning(f"Not found `{language}.so` in `{load_path}/tree-sitter/`, attemp to build language")
106106
build_language(language, load_path)
107-
language = Language(load_path + f"/tree-sitter/{language}.so", language)
108-
parser.set_language(language)
109-
107+
language = Language(load_path + f"/tree-sitter/{language}.so", language)
108+
parser.set_language(language)
109+
110110
if isinstance(raw_code, str):
111111
raw_code = bytes(raw_code, 'utf8')
112112
elif isinstance(raw_code, bytes):

0 commit comments

Comments
 (0)