fix test (#20191)

Co-authored-by: Wei Chu <[email protected]>

tests/python/unittest/test_operator.py

@@ -10041,10 +10041,7 @@ def test_scalarop_locale_invariance():
     finally:
         locale.setlocale(locale.LC_NUMERIC, prev)
 
-if __name__ == '__main__':
-    import nose
-    nose.runmodule()
-
+@with_seed()
 def test_take_grads():
     # Test for https://github.com/apache/incubator-mxnet/issues/19817
     from mxnet.gluon.nn import HybridBlock, Conv1D, HybridSequential, HybridLambda, Dense
@@ -10133,4 +10130,8 @@ def hybrid_forward(self, F, X, axis=-1):
     grads2 = run_model(model2, loss, X, Y)
 
     for i in range(len(grads1)):
-        assert_almost_equal(grads1[i], grads2[i])
+        assert_almost_equal(grads1[i], grads2[i])
+
+if __name__ == '__main__':
+    import nose
+    nose.runmodule()