def add_np_implementation(function, docstring): np_implementation = getattr(numpy_backend, function.__name__) code = inspect.getsource(np_implementation) code_lines = code.split('\n') for i in range(len(code_lines)): if code_lines[i]: # if there is something on the line, add 8 spaces. code_lines[i] = ' ' + code_lines[i] code = '\n'.join(code_lines[:-1]) if len(code_lines) < 10: section = template_np_implementation.replace('{{code}}', code) else: section = template_hidden_np_implementation.replace('{{code}}', code) return docstring.replace('{{np_implementation}}', section)
def add_np_implementation(function, docstring): np_implementation = getattr(numpy_backend, function.__name__) code = inspect.getsource(np_implementation) code_lines = code.split('\n') for i in range(len(code_lines)): if code_lines[i]: # if there is something on the line, add 8 spaces. code_lines[i] = ' ' + code_lines[i] code = '\n'.join(code_lines[:-1]) if len(code_lines) < 10: section = template_np_implementation.replace('{{code}}', code) else: section = template_hidden_np_implementation.replace('{{code}}', code) return docstring.replace('{{np_implementation}}', section)