[Bart] dont call .forward
@julien-c any idea why this would cause 
```
FAILED tests/test_pipelines.py::MonoColumnInputTestCase::test_tf_feature_extraction 
```<|||||>I think this one might sometimes fail randomly