Refactor inference manager code to save correct files

[Geeks-Sid]

Fixes #862

Proposed Changes

• Ensured that n_folds is not used to trigger logit saving

Checklist

• [x] CONTRIBUTING guide has been followed.
• [x] PR is based on the current GaNDLF master .
• [x] Non-breaking change (does not break existing functionality): provide as many details as possible for any breaking change.
• [x] Function/class source code documentation added/updated (ensure typing is used to provide type hints, including and not limited to using Optional if a variable has a pre-defined value).
• [x] Code has been blacked for style consistency and linting.
• [x] If applicable, version information has been updated in GANDLF/version.py.
• [x] If adding a git submodule, add to list of exceptions for black styling in pyproject.toml file.
• [x] Usage documentation has been updated, if appropriate.
• [ ] Tests added or modified to cover the changes; if coverage is reduced, please give explanation.
• [x] If customized dependency installation is required (i.e., a separate pip install step is needed for PR to be functional), please ensure it is reflected in all the files that control the CI, namely: python-test.yml, and all docker files [1,2,3].

[github-actions[bot]]

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 95.11%. Comparing base (50f369f) to head (bb8821d).

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #865 +/- ## ========================================== + Coverage 95.08% 95.11% +0.02% ========================================== Files 122 122 Lines 8342 8326 -16 ========================================== - Hits 7932 7919 -13 + Misses 410 407 -3 ``` | [Flag](https://app.codecov.io/gh/mlcommons/GaNDLF/pull/865/flags?src=pr&el=flags&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=mlcommons) | Coverage Δ | | |---|---|---| | [unittests](https://app.codecov.io/gh/mlcommons/GaNDLF/pull/865/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=mlcommons) | `95.11% <100.00%> (+0.02%)` | :arrow_up: | Flags with carried forward coverage won't be shown. [Click here](https://docs.codecov.io/docs/carryforward-flags?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=mlcommons#carryforward-flags-in-the-pull-request-comment) to find out more.

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.