/
githubmirror
/
Agent4Rec
Обзор
Документация
Войти
/
githubmirror
/
Agent4Rec
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
recommenders/setup.py
35 строк
919 B
chenyuxin1999
init repo
12 окт 2023, 05:43
12 окт 2023, 05:43
9a08cc7
Код
Авторство
О чём код?
from distutils.core import setup, Extension from Cython.Build import cythonize import numpy as np import os pyx_directories = ["evaluator/backend/cpp/", "util/cython"] cpp_dirs = ["evaluator/backend/cpp/include", "util/cython/include"] extensions = [ Extension( '*', ["*.pyx"], extra_compile_args=["-std=c++11"]) ] pwd = os.getcwd() additional_dirs = [os.path.join(pwd, d) for d in cpp_dirs] for t_dir in pyx_directories: target_dir = os.path.join(pwd, t_dir) os.chdir(target_dir) ori_files = set(os.listdir("./")) setup( ext_modules=cythonize(extensions, language="c++"), include_dirs=[np.get_include()]+additional_dirs ) new_files = set(os.listdir("./")) for n_file in new_files: if n_file not in ori_files and n_file.split(".")[-1] in ("c", "cpp"): os.remove(n_file) os.chdir(pwd)