diff --git a/.pylintrc b/.pylintrc
index 130124da70..a8308f4400 100644
--- a/.pylintrc
+++ b/.pylintrc
@@ -91,7 +91,6 @@ disable=
     no-self-use, # TODO: Convert methods to functions where appropriate
     consider-using-in, # TODO: Consider merging comparisons with "in"
     too-many-public-methods, # TODO: Resolve
-    ungrouped-imports, # TODO: Group imports
     consider-using-ternary, # TODO: Consider ternary expressions
     chained-comparison, # TODO: Simplify chained comparison between operands
     too-many-branches, # TODO: Simplify or ignore as appropriate
diff --git a/src/sagemaker/session.py b/src/sagemaker/session.py
index e0e30e66fc..b39e0be055 100644
--- a/src/sagemaker/session.py
+++ b/src/sagemaker/session.py
@@ -23,8 +23,8 @@
 import six
 import boto3
 import botocore.config
-import yaml
 from botocore.exceptions import ClientError
+import yaml
 
 import sagemaker.logs
 from sagemaker import vpc_utils