Replace subclassing with proper alias

zurutech · Aug 23, 2019 · 73e31b2 · 73e31b2
1 parent 1c3fed0
commit 73e31b2
Showing 1 changed file with 3 additions and 10 deletions.
diff --git a/ashpy/models/gans.py b/ashpy/models/gans.py
@@ -50,13 +50,6 @@
 from ashpy.models.convolutional.encoders import BaseEncoder
 
 
-class Generator(BaseDecoder):
-    """GAN Generator Model."""
-
-
-class Discriminator(BaseEncoder):
-    """GAN Discriminator Model."""
-
-
-class Encoder(BaseEncoder):
-    """GAN Encoder Model."""
+Generator = BaseDecoder
+Discriminator = BaseEncoder
+Encoder = BaseEncoder