ensure all args fp

apache · Jun 26, 2021 · f1ce2bb · f1ce2bb
1 parent ccb9e4b
commit f1ce2bb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
@@ -3367,8 +3367,8 @@ def test_lstm():
             hidden_size=32,
             use_bias=False,
             activations=["HardSigmoid", "LeakyRelu", "Tanh"] * directions,
-            alphas=[2.0, 0.5, 0] * directions,
-            betas=[0.3, 0, 0] * directions,
+            alphas=[2.0, 0.5, 0.0] * directions,
+            betas=[0.3, 0.0, 0.0] * directions,
             rnn_type="LSTM",
             directions=directions,
         )
@@ -3381,7 +3381,7 @@ def test_lstm():
             use_bias=False,
             activations=["HardSigmoid", "LeakyRelu", "Affine"] * directions,
             alphas=[2.0, 0.5, 0.8] * directions,
-            betas=[0.3, 0.1, 0] * directions,
+            betas=[0.3, 0.1, 0.0] * directions,
             rnn_type="LSTM",
             directions=directions,
         )
@@ -3497,7 +3497,7 @@ def test_gru():
             use_bias=False,
             activations=["HardSigmoid", "LeakyRelu"] * directions,
             alphas=[2.0, 0.5] * directions,
-            betas=[0.3, 0] * directions,
+            betas=[0.3, 0.0] * directions,
             rnn_type="GRU",
             directions=directions,
         )