fix: Refactor assertions in E2E tests for Dynamo#2001

peri044 merged 1 commit intomainpytorch/TensorRT:mainfrom

e2e_testing_small_fixpytorch/TensorRT:e2e_testing_small_fixCopy head branch name to clipboard