update import statements

bharathgs · Sep 13, 2018 · 0028086 · 0028086
1 parent 206ddff
commit 0028086
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/nalu/core/nac_cell.py b/nalu/core/nac_cell.py
@@ -1,7 +1,8 @@
 from torch import Tensor, nn
 from torch.nn.parameter import Parameter
 from torch.nn.init import xavier_uniform_
-from torch.nn.functional import tanh, sigmoid, linear
+from torch.nn.functional import linear
+from torch import sigmoid, tanh
 
 
 class NacCell(nn.Module):

diff --git a/nalu/core/nalu_cell.py b/nalu/core/nalu_cell.py
@@ -1,7 +1,8 @@
 from torch import Tensor, exp, log, nn
 from torch.nn.parameter import Parameter
 from torch.nn.init import xavier_uniform_
-from torch.nn.functional import sigmoid, linear
+from torch.nn.functional import linear
+from torch import sigmoid
 from .nac_cell import NacCell