fixed names

This commit is contained in:
Kashif Rasul
2022-04-29 13:28:08 +02:00
parent 20217bc9c8
commit 1bbb48c6a3
+3 -3
View File
@@ -3,7 +3,7 @@ from .lightning_module import TransformerLightningModule
from .module import TransformerModel
__all__ = [
"TransformerModel",
"TransformerLightningModule",
"TransformerEstimator",
"ReformerModel",
"ReformerLightningModule",
"ReformerEstimator",
]