diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks.py b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks.py
index 555de78b..a261c8d1 100644
--- a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks.py
+++ b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks.py
@@ -18,8 +18,6 @@
 from __future__ import division
 from __future__ import print_function
 
-# import g3
-
 import numpy as np
 import six
 import tensorflow as tf
diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_impl_test.py b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_impl_test.py
index d8b98e8a..ca72d9c8 100644
--- a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_impl_test.py
+++ b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_impl_test.py
@@ -18,8 +18,6 @@
 from __future__ import division
 from __future__ import print_function
 
-# import g3
-
 from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_utils.py b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_utils.py
index c92ff52c..45379dd6 100644
--- a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_utils.py
+++ b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_utils.py
@@ -20,7 +20,6 @@
 from __future__ import division
 from __future__ import print_function
 
-# import g3
 import numpy as np
 import tensorflow as tf
 
diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_utils_test.py b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_utils_test.py
index dcdb36eb..7ec44d53 100644
--- a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_utils_test.py
+++ b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_utils_test.py
@@ -18,8 +18,6 @@
 from __future__ import division
 from __future__ import print_function
 
-# import g3
-
 from absl.testing import parameterized
 import tensorflow as tf
 
diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_wrapper.py b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_wrapper.py
index 02fa34a7..e7c81a8a 100644
--- a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_wrapper.py
+++ b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_wrapper.py
@@ -22,8 +22,6 @@
 import inspect
 import re
 
-# import g3
-
 import numpy as np
 import tensorflow as tf