Fix failing torchscript tests for `CpmAnt` model 
# What does this PR do?

The failure is caused by some type issues (dict, tuple, etc) in the outputs.
_The documentation is not available anymore as the PR was closed or merged._