Make the utils.isnan function inline (jmschrei#624)

pepie34 · Oct 1, 2019 · 2dc0f9d · 2dc0f9d
1 parent 52e7d30
commit 2dc0f9d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/pomegranate/utils.pxd b/pomegranate/utils.pxd
@@ -8,7 +8,9 @@ cdef int* GPU
 cdef extern from "numpy/npy_math.h":
 	bint npy_isnan(double x) nogil
 
-cdef bint isnan(double x) nogil
+cdef inline bint isnan(double x) nogil:
+	return npy_isnan(x)
+
 cdef int _is_gpu_enabled() nogil
 cdef python_log_probability(model, double* X, double* log_probability, int n)
 cdef python_summarize(model, double* X, double* weights, int n)

diff --git a/pomegranate/utils.pyx b/pomegranate/utils.pyx
@@ -44,9 +44,6 @@ numpy.import_array()
 cdef extern from "numpy/ndarraytypes.h":
 	void PyArray_ENABLEFLAGS(numpy.ndarray X, int flags)
 
-cdef bint isnan(double x) nogil:
-	return npy_isnan(x)
-
 # Define some useful constants
 DEF NEGINF = float("-inf")
 DEF INF = float("inf")
@@ -487,4 +484,4 @@ def check_random_state(seed):
 		return seed
 	raise ValueError('%r cannot be used to seed a numpy.random.RandomState'
 					 ' instance' % seed)
-
+