fix bug in numba tools
import numpy as np
-from numba import njit, vectorize
+from numba import njit
from HARK.rewards import (
CRRAutility,
@@ -11,13 +11,13 @@ from HARK.rewards import (
CRRAutilityPP,
)
-CRRAutility = vectorize(CRRAutility, cache=True)
-CRRAutilityP = vectorize(CRRAutilityP, cache=True)
-CRRAutilityPP = vectorize(CRRAutilityPP, cache=True)
-CRRAutilityP_inv = vectorize(CRRAutilityP_inv, cache=True)
-CRRAutility_invP = vectorize(CRRAutility_invP, cache=True)
-CRRAutility_inv = vectorize(CRRAutility_inv, cache=True)
-CRRAutilityP_invP = vectorize(CRRAutilityP_invP, cache=True)
+CRRAutility = njit(CRRAutility, cache=True)
+CRRAutilityP = njit(CRRAutilityP, cache=True)
+CRRAutilityPP = njit(CRRAutilityPP, cache=True)
+CRRAutilityP_inv = njit(CRRAutilityP_inv, cache=True)
+CRRAutility_invP = njit(CRRAutility_invP, cache=True)
+CRRAutility_inv = njit(CRRAutility_inv, cache=True)
+CRRAutilityP_invP = njit(CRRAutilityP_invP, cache=True)
@njit(cache=True, error_model="numpy")
