Add import sagemaker_containers to mnist user script to keep it more consistent with blog post.

Author: nadiaya

sagemaker-python-sdk/pytorch_mnist/mnist.py

@@ -2,6 +2,7 @@
 import json
 import logging
 import os
+import sagemaker_containers
 import sys
 import torch
 import torch.distributed as dist