numba - TypingError: cannot determine Numba type of <class 'builtin_function_or_method'>

CIsForCookies picture CIsForCookies · Jun 7, 2018 · Viewed 28.4k times · Source

I have a simple function to rank poker hands (the hands are strings).

I call it with rA,rB = rank(a),rank(b) and here is my implementation. Works well without @jit(nopython=True), but with it, it fails:

   File "C:/Users/avi_na/Desktop/poker.py", line 190, in <module>
        rA,rB = rank(a),rank(b)

      File "C:\Continuum\anaconda3\lib\site-packages\numba\dispatcher.py", line 344, in _compile_for_args
        reraise(type(e), e, None)

      File "C:\Continuum\anaconda3\lib\site-packages\numba\six.py", line 658, in reraise
        raise value.with_traceback(tb)

TypingError: cannot determine Numba type of <class 'builtin_function_or_method'>

from numba import jit
from numba.types import string

@jit(nopython=True)
def rank(hand):
#    assert(len(hand) == 5)
    rank = "N/A"

    p = pd.Series([h[0] for h in hand]).value_counts()
    v = sorted(set(pd.Series([h[0] for h in hand]).values), reverse=True)
    s = sorted(hand, key=lambda k:k[0])    
    z = zip(s,s[1:])

    if all(x[0]==y[0]-1 for x,y in z):
        rank = "Straight "

    if len(set([h[1] for h in hand])) == 1:
        rank += "Flush "

    if "Straight Flush" in rank and sum([h[0] for h in hand]) == sum([10,11,12,13,14]):
        rank = "Royal Flush"

    elif p[p.idxmax()] == 4:
        rank = "4 Of A Kind : %d" % p.idxmax()

    elif p[p.idxmax()] == 3 and p[p.idxmin()] == 1:
        rank = "3 Of A Kind : %d" % p.idxmax()

    elif p[p.idxmax()] == 3 and p[p.idxmin()] == 2:
        rank = "Full House : %d,%d" % (p.idxmax(), p.idxmin())

    elif p[p.idxmax()] == 2:
        max2 = p.nlargest(2)

        if list(max2) == [2,2]:
            max2 = sorted(list(max2.keys()), reverse=True)
            rank = "2 Pairs : %d,%d" % (max2[0],max2[1])
        else:
            rank = "Pair : %d" % p.idxmax()

    else:
        rank = "High Card : %d" % v[0]


    return rank

Answer

Kevin K. picture Kevin K. · Jun 7, 2018

Pandas and several other function calls in your code will not work with nopython=True. The available libraries that can be used with numba jit in nopython is fairly limited (pretty much only to numpy arrays and certain python builtin libraries). You can find more information here