lint

Dmytro Dzhulgakov · Dmytro Dzhulgakov · commit 4e94499f6b12 · 2019-06-20T13:57:15.000-07:00
diff --git a/test/test_nn.py b/test/test_nn.py
@@ -4675,7 +4675,7 @@ class CustomState(nn.Module):
             def __init__(self):
                 super(CustomState, self).__init__()
                 self.param = torch.nn.Parameter(torch.ones(1))
-                self.sub = torch.nn.Linear(5,5)
+                self.sub = torch.nn.Linear(5, 5)
 
             def _save_to_state_dict(self, destination, prefix, keep_vars):
                 destination[prefix + "serialized"] = self.param.data + 1
