diff --git a/privacy/bolton/__init__.py b/privacy/bolton/__init__.py
index d3c8209..1660208 100644
--- a/privacy/bolton/__init__.py
+++ b/privacy/bolton/__init__.py
@@ -23,7 +23,7 @@ if LooseVersion(tf.__version__) < LooseVersion("2.0.0"):
 if hasattr(sys, "skip_tf_privacy_import"):  # Useful for standalone scripts.
   pass
 else:
-  from privacy.bolton.models import BoltonModel
-  from privacy.bolton.optimizers import Bolton
-  from privacy.bolton.losses import StrongConvexHuber
-  from privacy.bolton.losses import StrongConvexBinaryCrossentropy
+  from privacy.bolton.models import BoltonModel  # pylint: disable=import-not-at-top
+  from privacy.bolton.optimizers import Bolton  # pylint: disable=import-not-at-top
+  from privacy.bolton.losses import StrongConvexHuber  # pylint: disable=import-not-at-top
+  from privacy.bolton.losses import StrongConvexBinaryCrossentropy  # pylint: disable=import-not-at-top
diff --git a/privacy/bolton/losses_test.py b/privacy/bolton/losses_test.py
index 713bd7c..3996ae9 100644
--- a/privacy/bolton/losses_test.py
+++ b/privacy/bolton/losses_test.py
@@ -17,8 +17,8 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-from contextlib import contextmanager
-from io import StringIO
+from contextlib import contextmanager  # pylint: disable=importing-member
+from io import StringIO  # pylint: disable=importing-member
 import sys
 from absl.testing import parameterized
 import tensorflow as tf