From 319bf81bb166f4c43ca898a509bfdc56e1558280 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Tue, 28 Apr 2020 00:37:35 -0700
Subject: [PATCH] Setting correct argument in super() call in DPOptimizerClass.

PiperOrigin-RevId: 308776729
---
 tensorflow_privacy/privacy/optimizers/dp_optimizer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow_privacy/privacy/optimizers/dp_optimizer.py b/tensorflow_privacy/privacy/optimizers/dp_optimizer.py
index 210ade8..3706bf0 100644
--- a/tensorflow_privacy/privacy/optimizers/dp_optimizer.py
+++ b/tensorflow_privacy/privacy/optimizers/dp_optimizer.py
@@ -135,7 +135,7 @@ def make_optimizer_class(cls):
 
         def process_microbatch(i, sample_state):
           """Process one microbatch (record) with privacy helper."""
-          grads, _ = zip(*super(cls, self).compute_gradients(
+          grads, _ = zip(*super(DPOptimizerClass, self).compute_gradients(
               tf.reduce_mean(input_tensor=tf.gather(
                   microbatches_losses, [i])), var_list, gate_gradients,
               aggregation_method, colocate_gradients_with_ops, grad_loss))