Fix lint

Author: leezu

gluonnlp/data/candidate_sampler.py

@@ -24,11 +24,9 @@
 import numpy as np
 
 try:
-    from numba import njit, prange
+    from numba import njit
     numba_njit = njit(nogil=True)
 except ImportError:
-    # Define numba shims
-    prange = range
 
     def numba_njit(func):
         return func

scripts/word_embeddings/train_fasttext.py

@@ -17,7 +17,7 @@
 # specific language governing permissions and limitations
 # under the License.
 
-# pylint: disable=global-variable-undefined
+# pylint: disable=global-variable-undefined,wrong-import-position
 """Fasttext embedding model
 ===========================
 
@@ -330,7 +330,7 @@ def train(args):
                     if args.no_deduplicate_words:
                         subwords, subwords_mask = \
                             indices_to_subwordindices_mask(word_context,
-                                                       idx_to_subwordidxs)
+                                                           idx_to_subwordidxs)
                     else:
                         unique, inverse_unique_indices = np.unique(
                             word_context.asnumpy(), return_inverse=True)
@@ -460,8 +460,7 @@ def train(args):
         mx.nd.waitall()
     with print_time('evaluate'):
         evaluate(args, embedding, vocab, num_update,
-                 eval_analogy=(epoch == args.epochs - 1
-                               and not args.no_eval_analogy))
+                 eval_analogy=not args.no_eval_analogy)
 
     # Save params
     with print_time('save parameters'):

scripts/word_embeddings/utils.py

@@ -27,7 +27,6 @@
 import random
 
 import mxnet as mx
-import numpy as np
 
 try:
     from numba import njit

tests/unittest/test_sampler.py

@@ -73,6 +73,6 @@ def test_context_sampler():
 
     center, context, mask = next(iter(sampler))
 
-    assert center.asnumpy().tolist() == [0, 1]
+    assert center.asnumpy().tolist() == [[0, 1]]
     assert context.asnumpy().tolist() == [[1, 0], [0, 2]]
     assert mask.asnumpy().tolist() == [[1, 0], [1, 1]]

tests/unittest/test_vocab_embed.py

@@ -952,5 +952,5 @@ def test_subword_function_ngramhashes():
 
     assert [8.0, 195.0, 271.0, 500.0, 201.0, 445.0, 379.0, 831.0, 617.0, 851.0] == \
         sf([u'test'])[0].asnumpy().tolist()
-    assert [253.0, 801.0, 557.0, 966.0, 63.0, 874.0, 313.0, 188.0, 588.0, 86.0] == \
+    assert [429.0, 793.0, 101.0, 334.0, 295.0, 474.0, 145.0, 524.0, 388.0, 790.0] == \
         sf([u'τεστ'])[0].asnumpy().tolist()