Need help with displaying my my predictions.

[memora0101]

Hello I am currently struggling with printing the sentence along with the 2 output prediction for the sentence.

The following is the issues I am facing with:

for sent in train_x:
    inputs = np.array(sent_to_tensor(sent, vocab_dict=Vocab))
    inputs = inputs[None, :]
    example_input = next(val_generator(batch_size=batch_size, shuffle=True))
    tmp_inputs, tmp_targets, tmp_example_weights = example_input
    tmp_pred = training_loop.eval_model(tmp_inputs)
    print(f'example input_str: {sent}')
    print(f'Model returned sentiment probabilities: {tmp_pred[sent]}')

example input_str: Hello, is this apple product support?

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-28-65441ecee974> in <module>
      6     tmp_pred = training_loop.eval_model(tmp_inputs)
      7     print(f'example input_str: {sent}')
----> 8     print(f'Model returned sentiment probabilities: {tmp_pred[sent]}')

~/opt/anaconda3/lib/python3.7/site-packages/jax/_src/numpy/lax_numpy.py in _rewriting_take(arr, idx)
   4244   # followed by an optional reverse and broadcast_in_dim.
   4245   arr = asarray(arr)
-> 4246   treedef, static_idx, dynamic_idx = _split_index_for_jit(idx)
   4247   return _gather(arr, treedef, static_idx, dynamic_idx)
   4248 

~/opt/anaconda3/lib/python3.7/site-packages/jax/_src/numpy/lax_numpy.py in _split_index_for_jit(idx)
   4300   """
   4301   # Convert list indices to tuples in cases (deprecated by NumPy.)
-> 4302   idx = _eliminate_deprecated_list_indexing(idx)
   4303 
   4304   # Expand any (concrete) boolean indices. We can then use advanced integer

~/opt/anaconda3/lib/python3.7/site-packages/jax/_src/numpy/lax_numpy.py in _eliminate_deprecated_list_indexing(idx)
   4549                "use `arr[array(seq)]` instead of `arr[seq]`. "
   4550                "See https://github.com/google/jax/issues/4564 for more information.")
-> 4551       raise TypeError(msg)
   4552     else:
   4553       idx = (idx,)

TypeError: Using a non-tuple sequence for multidimensional indexing is not allowed; use `arr[tuple(seq)]` instead of `arr[seq]`. See https://github.com/google/jax/issues/4564 for more information.