Skip to content
Snippets Groups Projects
Commit 73101651 authored by ferrari's avatar ferrari
Browse files

Auto compile c_corr

parent ac1857d8
No related branches found
No related tags found
No related merge requests found
from distutils.core import setup
from Cython.Build import cythonize
setup(
ext_modules = cythonize("c_corr.pyx", annotate=True, compiler_directives={'language_level' : "3"})
)
\ No newline at end of file
setup(ext_modules=cythonize("c_corr.pyx", annotate=True, compiler_directives={'language_level': "3"}))
import argparse
import itertools
import os
import sys
import itertools
import argparse
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import PolynomialFeatures
from sklearn.linear_model import LinearRegression
import numpy as np
from numpy.fft import rfft, irfft
import scipy.signal as sg
import soundfile as sf
import c_corr
from gsrp_smart_util import *
from math import ceil
from numpy.fft import rfft, irfft
from scipy.signal.windows import tukey
from sklearn.linear_model import LinearRegression
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import PolynomialFeatures
try:
from tqdm import trange
except ImportError:
trange = range
from gsrp_smart_util import *
class BColors:
......@@ -32,6 +26,20 @@ class BColors:
UNDERLINE = '\033[4m'
try:
import c_corr
except ModuleNotFoundError:
print(f'{BColors.WARNING}Unable to load c_corr. Attempting to recompile...{BColors.ENDC}')
from subprocess import run
run('python cython_setup.py build_ext --inplace'.split())
import c_corr
print(f'{BColors.OKGREEN}Done{BColors.ENDC}')
try:
from tqdm import trange
except ImportError:
trange = range
def intlist(s):
return list(map(int, s.split(',')))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment