Fix style checks for DGM poisoning attacks

Beat Buesser · Beat Buesser · commit 3569d8523bd0 · 2022-05-05T14:21:11.000+01:00
Signed-off-by: Beat Buesser &lt;beat.buesser@ie.ibm.com&gt;
diff --git a/tests/utils.py b/tests/utils.py
@@ -426,7 +426,7 @@ def discriminator_loss_fct(real_output, generated_output):
 
         return total_loss
 
-    gan = TensorFlow2GAN(
+    gan = TensorFlowV2GAN(
         generator=generator,
         discriminator=discriminator_classifier,
         generator_loss=generator_orig_loss_fct,
