Adding function to skip test for py2 verison (#128)

Author: Roshrini

CHANGELOG.md

@@ -79,3 +79,4 @@
  * Add support for fastai (https://github.com/fastai/fastai) library.
  * Remove "requsests" from tests dependencies to avoid regular conflicts with "requests" package from "sagemaker" dependencies.
  * Add support for PyTorch-1.0.
+ 

test/conftest.py

@@ -190,6 +190,12 @@ def skip_by_device_type(request, use_gpu, instance_type):
         pytest.skip('Skipping because running on \'{}\' instance'.format(instance_type))
 
 
+@pytest.fixture(autouse=True)
+def skip_by_py_version(request, py_version):
+    if request.node.get_closest_marker('skip_py2') and py_version != 'py3':
+        pytest.skip('Skipping the test because Python 2 is not supported.')
+
+
 @pytest.fixture(autouse=True)
 def skip_gpu_instance_restricted_regions(region, instance_type):
     if((region in NO_P2_REGIONS and instance_type.startswith('ml.p2')) or