[test] Fix smclarify test flakiness#1082

saimidu merged 2 commits intoaws:masteraws/deep-learning-containers:masterfrom

saimidu:fix_smclarify_testsaimidu/deep-learning-containers:fix_smclarify_testCopy head branch name to clipboard