Update tests for Nan Regression
Fixes some issues in !141 (merged).
- Adds nan_regression unit test to gitlab cli.
- Fixes regression task in cases where some labels are NaN. Previous approach seems not to work, due to some weird inner workings of pytorch.
Edited by Nikita Ivvan Pond