fixing init

pyg-team · Dec 15, 2024 · a37e0bb · a37e0bb
1 parent c525a42
commit a37e0bb
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/torch_geometric/contrib/nn/models/__init__.py b/torch_geometric/contrib/nn/models/__init__.py
@@ -1,5 +1,3 @@
 from .rbcd_attack import PRBCDAttack, GRBCDAttack
 
-__all__ = classes = [
-    'PRBCDAttack', 'GRBCDAttack', 'GeneralizedGenerator', 'HeGAN'
-]
+__all__ = classes = ['PRBCDAttack', 'GRBCDAttack', 'HeGAN']