Fixing a bug when MlFlow try to log a torch.tensor

[etiennebonnafoux]

What does this PR do?

Add the case where a torch.tensor with one element is log on Mlflow. For now it only accept float and int.

Fixes # (issue)

Before submitting

• This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
• Did you read the contributor guideline,
  Pull Request section?
• Was this discussed/approved via a Github issue or the forum? Please add a link
  to it if that's the case.
  Here Trainer is attempting to log a torch.Tensor but MLflow's only accepts float. #29784
• Did you make sure to update the documentation with your changes? Here are the
  documentation guidelines, and
  here are tips on formatting docstrings.
  No need
• Did you write any new necessary tests?
  No need (two lines of code)

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

@amyeroberts

[amyeroberts]

Thanks for fixing! Just a small suggestion

[etiennebonnafoux]

Sorry but what should I do to pass check_code_quality and setup_and_quality test ? this is my first PR

[amyeroberts]

@etiennebonnafoux Running make fixup and pushing the changes should resolve this. There's more info in the contributing guideline