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