fix: adjust assertion of tensorflow mnist test#2003

ajaykarpur merged 3 commits intoaws:masteraws/sagemaker-python-sdk:masterfrom

metrizable:fix-flaky-testmetrizable/sagemaker-python-sdk:fix-flaky-testCopy head branch name to clipboard