def predict()

in notebooks/src/transformscript.py [0:0]

• 9 lines of code
• 4 McCabe index (conditional complexity)

def predict(model, payload):
    try:
        # locally, payload may come in as an np.ndarray
        if type(payload)==np.ndarray:
            out = [str(model.predict(payload.reshape((1,8))))]
        # in remote / container based deployment, payload comes in as a stream of bytes
        else:

            out = [str(model.predict(np.frombuffer(payload).reshape((1,8))))]
    except Exception as e:
        out = [type(payload),str(e)] #useful for debugging!
    
    return out