Simplified calls to superclass constructors.

[pytorch.git] / lazy_linear.py

diff --git a/lazy_linear.py b/lazy_linear.py
index 2fa6a29..97530ef 100755 (executable)
--- a/lazy_linear.py
+++ b/lazy_linear.py
@@ -12,7 +12,7 @@ from torch import nn, Tensor
 class LazyLinear(nn.Module):
 
     def __init__(self, out_dim, bias = True):
-        super(LazyLinear, self).__init__()
+        super().__init__()
         self.out_dim = out_dim
         self.bias = bias
         self.core = None
@@ -30,7 +30,7 @@ class LazyLinear(nn.Module):
 
     def named_parameters(self, memo=None, prefix=''):
         assert self.core is not None, 'Parameters not yet defined'
-        return super(LazyLinear, self).named_parameters(memo, prefix)
+        return super().named_parameters(memo, prefix)
 
 ######################################################################