distributed_train_test_step_fns is not being graded. And it shows There was a problem grading your submission. Check stderr for more details.
But everything seems right. Can someone help me on this.
Are you still having the problem? I just uploaded my submission, and it was graded without any issues whatsoever.