message: Keras subclass model wrapper for Graph Regularization.
diff: @@ -12,6 +12,7 @@ visibility = [
  "//tensorflow_models:__subpackages__",
  "//tensorflow_model_optimization:__subpackages__",
  "//third_party/py/cleverhans:__subpackages__",
  + "//third_party/py/neural_structured_learning/keras:__pkg__",
  "//third_party/py/tensorflow_examples:__subpackages__",
  "//third_party/py/tf_slim:__subpackages__",
  # TODO(aselle): to pass open source test.
  