Fixed bug in inference

[vahluw]

…2D patches

Fixes #ISSUE_NUMBER

Proposed Changes

• When using a 2D patch such as (2048, 2048), during validation the patch shape is (2048, 2048, 1). This triggers the else statement in forward_pass.py, causing an issue when trying to squeeze().

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.
• 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).
• Code has been blacked for style consistency and linting.
• If applicable, version information has been updated in GANDLF/version.py.
• If adding a git submodule, add to list of exceptions for black styling in pyproject.toml file.
• 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]

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

[vahluw]

recheck

[Geeks-Sid]

Great catch