Dear cmubig:
I'm very interested in your work, it's a very interesting work to do trajectory prediction in a different perspective, I'm encountering an error while debugging the code, AttributeError: 'DataFrame' object has no attribute 'orig_idx'. Can you help me to solve this problem?

Best Regards