Fix regression test failures

[namannandan]

Description

Fix regression test failures associated with kserve PR: #2175
Breaking change: https://github.com/pytorch/serve/pull/2175/files#diff-57322aac10a7e3d1c0a846bba41af9ca7947ec12bf05f0a27c4a080cd885ac4aR170

Fixes #2366

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)

Feature/Issue validation/testing

Please describe the Unit or Integration tests that you ran to verify your changes and relevant result summary. Provide instructions so it can be reproduced.
Please also list any relevant details for your test configuration.

• Sanity and regression tests in CI

[codecov]

Codecov Report

Merging #2371 (8c84742) into master (81c3d59) will not change coverage.
The diff coverage is n/a.

❗ Current head 8c84742 differs from pull request most recent head 3810ff9. Consider uploading reports for the commit 3810ff9 to get more accurate results

@@           Coverage Diff           @@
##           master    #2371   +/-   ##
=======================================
  Coverage   72.15%   72.15%           
=======================================
  Files          78       78           
  Lines        3641     3641           
  Branches       58       58           
=======================================
  Hits         2627     2627           
  Misses       1010     1010           
  Partials        4        4           

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more