Change assert_equal to assert_almost_equal

Author: ryan.smith

test/classification/test_loss.py

@@ -19,15 +19,15 @@ def test_sce_equals_ce(self):
 
         ce_loss = F.cross_entropy(Y_probs, Y_golds, reduction="none")
         ces_loss = cross_entropy_with_probs(Y_probs, Y_golds_probs, reduction="none")
-        np.testing.assert_equal(ce_loss.numpy(), ces_loss.numpy())
+        np.testing.assert_almost_equal(ce_loss.numpy(), ces_loss.numpy(), decimal=6)
 
         ce_loss = F.cross_entropy(Y_probs, Y_golds, reduction="sum")
         ces_loss = cross_entropy_with_probs(Y_probs, Y_golds_probs, reduction="sum")
-        np.testing.assert_equal(ce_loss.numpy(), ces_loss.numpy())
+        np.testing.assert_almost_equal(ce_loss.numpy(), ces_loss.numpy(), decimal=6)
 
         ce_loss = F.cross_entropy(Y_probs, Y_golds, reduction="mean")
         ces_loss = cross_entropy_with_probs(Y_probs, Y_golds_probs, reduction="mean")
-        np.testing.assert_equal(ce_loss.numpy(), ces_loss.numpy())
+        np.testing.assert_almost_equal(ce_loss.numpy(), ces_loss.numpy(), decimal=6)
 
     def test_perfect_predictions(self):
         # Does soft ce loss achieve approx. 0 loss with perfect predictions?