How to pass Sparse Tensors into a Keras Model on SageMaker

Question

I have tried to make the example as minimal as possible. The following script creates a SparseTensor similiar to my input:

import numpy as np
import tensorflow as