From aedd0cf4250174926ff8b0c7ae4ea6849c135ce1 Mon Sep 17 00:00:00 2001 From: Amir MOHAMMADI <amir.mohammadi@idiap.ch> Date: Tue, 13 Jul 2021 17:56:16 +0200 Subject: [PATCH] [_library.py] implement array_jit decorator correctly Fixes #64 --- bob/measure/_library.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/bob/measure/_library.py b/bob/measure/_library.py index 4fce552..d1b2ba4 100644 --- a/bob/measure/_library.py +++ b/bob/measure/_library.py @@ -10,6 +10,7 @@ import numpy import numpy.linalg from numba import jit, objmode import logging +from functools import wraps logger = logging.getLogger(__name__) @@ -26,6 +27,7 @@ def _lists_to_arrays(*args, **kwargs): def array_jit(func): jit_func = jit(func, nopython=True) + @wraps(jit_func) def new_func(*args, **kwargs): args, kwargs = _lists_to_arrays(*args, **kwargs) return jit_func(*args, **kwargs) -- GitLab