remove wrong annotation (#2349)

2020-06-25 04:29:26 +02:00 · 2020-06-25 04:29:26 +02:00 · 220bb6db57
parent 9b2e60530f
commit 220bb6db57
1 changed files with 1 additions and 1 deletions
--- a/pl_examples/models/lightning_template.py
+++ b/pl_examples/models/lightning_template.py
@ -46,7 +46,7 @@ class LightningTemplateModel(LightningModule):
                 out_features: int = 10,
                 hidden_dim: int = 1000,
                 **kwargs
-                 ) -> 'LightningTemplateModel':
+                 ):
        # init superclass
        super().__init__()
        self.drop_prob = drop_prob