Use test dataset mask for prediction output

[yahelnachum]

Resolution Table

Date	Branch	Link	Details
-	-	-	-
-	-	-	-

---

Problem

Based on our last group meeting w/ the notebook team, they mentioned a mask that is given to tell us which test agents we are required to predict 50 future frames for. Right now our notebook pull all test agents. Instead we should use the mask to only predict future frames for ones required by the competition.

Solution

Same as above.

[yahelnachum]

10-22-2020

• using mask to only look at test dataset that we need to predict for