It would be nice to use directly np or tf types to define `in_tensor_dtype`. eg: ``` {'in_tensor_name': 'inputs', 'in_tensor_dtype': np.float32, 'data': np.random.rand(10,2)} ``` Also, what about make float the default type? So if somebody is lazy it can just not pass `in_tensor_dtype`? Btw, awesome work! You save my day :) Cheers, Francesco Saverio