From 1bbb48c6a3f26f46358abe1d2a836e5e34815908 Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@gmail.com>
Date: Fri, 29 Apr 2022 13:28:08 +0200
Subject: [PATCH] fixed names

---
 reformer/__init__.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/reformer/__init__.py b/reformer/__init__.py
index 0f8224e..1c557ba 100644
--- a/reformer/__init__.py
+++ b/reformer/__init__.py
@@ -3,7 +3,7 @@ from .lightning_module import TransformerLightningModule
 from .module import TransformerModel
 
 __all__ = [
-    "TransformerModel",
-    "TransformerLightningModule",
-    "TransformerEstimator",
+    "ReformerModel",
+    "ReformerLightningModule",
+    "ReformerEstimator",
 ]