Add __ALL__ to ashpy.models.gans

zurutech · Jan 30, 2020 · 7dc0b40 · 7dc0b40
1 parent 7c44d35
commit 7dc0b40
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/ashpy/models/gans.py b/src/ashpy/models/gans.py
@@ -54,6 +54,15 @@
 from ashpy.models.fc.decoders import Decoder as DenseDecoder
 from ashpy.models.fc.encoders import Encoder as DenseEncoder
 
+__ALL__ = [
+    "ConvGenerator",
+    "ConvDiscriminator",
+    "ConvEncoder",
+    "DenseGenerator",
+    "DenseDiscriminator",
+    "DenseEncoder",
+]
+
 ConvGenerator = ConvDecoder
 ConvDiscriminator = ConvEncoder
 ConvEncoder = ConvEncoder