From 121982deb156adc66e7b634a8a6bf31cb307a75a Mon Sep 17 00:00:00 2001
From: Steve Chien <schien@google.com>
Date: Thu, 8 Apr 2021 14:51:57 -0700
Subject: [PATCH] Add explicit package names to class __doc__ strings for
 DNNClassifier classes.

PiperOrigin-RevId: 367512322
---
 tensorflow_privacy/privacy/estimators/dnn.py    | 2 +-
 tensorflow_privacy/privacy/estimators/v1/dnn.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow_privacy/privacy/estimators/dnn.py b/tensorflow_privacy/privacy/estimators/dnn.py
index cac6034..0190c57 100644
--- a/tensorflow_privacy/privacy/estimators/dnn.py
+++ b/tensorflow_privacy/privacy/estimators/dnn.py
@@ -25,7 +25,7 @@ from tensorflow_estimator.python.estimator.canned import dnn
 
 
 class DNNClassifier(tf.estimator.Estimator):
-  """DP version of DNNClassifier."""
+  """DP version of `tf.estimator.DNNClassifier`."""
 
   def __init__(
       self,
diff --git a/tensorflow_privacy/privacy/estimators/v1/dnn.py b/tensorflow_privacy/privacy/estimators/v1/dnn.py
index 8ad1b94..1f480ef 100644
--- a/tensorflow_privacy/privacy/estimators/v1/dnn.py
+++ b/tensorflow_privacy/privacy/estimators/v1/dnn.py
@@ -22,7 +22,7 @@ from tensorflow_estimator.python.estimator.canned import dnn
 
 
 class DNNClassifier(tf.estimator.Estimator):
-  """DP version of tf.estimator.DNNClassifier."""
+  """DP version of `tf.compat.v1.estimator.DNNClassifier`."""
 
   def __init__(
       self,