Fix tests error

setup.py

@@ -14,7 +14,6 @@
     url="https://github.com/guocheng2018/transformer-encoder",
     packages=setuptools.find_packages(),
     python_requires=">=3.5",
-    install_requires=["pytorch>=1.0.0"],
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License",

tests/test_encoder.py

@@ -12,7 +12,7 @@
 
 
 def test_encoder():
-    enc = TransformerEncoder(n_layers, d_model, d_ff, n_heads, dropout)
+    enc = TransformerEncoder(d_model, d_ff, n_heads=n_heads, n_layers=n_layers, dropout=dropout)
     x = torch.randn(batch_size, max_len, d_model)
     mask = torch.randn(batch_size, max_len).ge(0)
     out = enc(x, mask)

tests/test_optim.py

@@ -9,12 +9,12 @@
 dropout = 0.1
 n_layers = 6
 
-factor = 1
-warmup = 20
+scale_factor = 1
+warmup_steps = 20
 
 
 def test_optim():
-    enc = TransformerEncoder(n_layers, d_model, d_ff, n_heads, dropout)
-    opt = WarmupOptimizer(d_model, factor, warmup, optim.Adam(enc.parameters()))
+    enc = TransformerEncoder(d_model, d_ff, n_heads=n_heads, n_layers=n_layers, dropout=dropout)
+    opt = WarmupOptimizer(optim.Adam(enc.parameters()), d_model, scale_factor, warmup_steps)
     assert type(opt.rate(step=1)) is float  # step starts from 1
     opt.step()

tests/test_pe.py

@@ -9,7 +9,7 @@
 
 
 def test_pe():
-    PE = PositionalEncoding(d_model, dropout, max_len)
+    PE = PositionalEncoding(d_model, dropout=dropout, max_len=max_len)
     embeds = torch.randn(batch_size, max_len, d_model)  # (batch_size, max_len, d_model)
     out = PE(embeds)
     assert embeds.size() == out.size()