scikit_learn_iris.py script embedded in notebook does not import tree

The Jupyter notebook (c80657daa9d42b7c9b12729d6fa4b825fd980730) for the scikit-learn estimator example contains an embedded example of the Python script `scikit_learn_iris.py`. The [script provided in the repo](https://github.com/awslabs/amazon-sagemaker-examples/blob/master/sagemaker-python-sdk/scikit_learn_iris/scikit_learn_iris.py) is correct, but the [version in the notebook](https://github.com/awslabs/amazon-sagemaker-examples/blob/master/sagemaker-python-sdk/scikit_learn_iris/Scikit-learn%20Estimator%20Example%20With%20Batch%20Transform.ipynb) is not - specifically, the commands in the notebook is missing

```python
import sklearn.tree
```

Instead, it imports `sklearn.svm`, which is not used in the script.