torch tensor type error with mypy #5563

Expected Behavior

Current Behavior

feast/online_response.py:104: error: Variable "feast.online_response.TorchTensor" is not valid as a type [valid-type]

Steps to reproduce

Specifications

• Version:
• Platform:
• Subsystem:

Possible Solution