Update tests/integ/test_xgboost.py

Author: verdimrc

tests/integ/test_xgboost.py

@@ -13,7 +13,7 @@
 from __future__ import absolute_import
 
 import os
-
+import pytest
 from sagemaker.xgboost.processing import XGBoostProcessor
 from tests.integ import DATA_DIR, TRAINING_DEFAULT_TIMEOUT_MINUTES
 from tests.integ.timeout import timeout