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