Passing target_ios correctly to test_gru (#477)

onnx · Sep 17, 2019 · f0a9327 · f0a9327
1 parent 1cccc9a
commit f0a9327
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/pytorch_model_test.py b/tests/pytorch_model_test.py
@@ -261,7 +261,7 @@ def forward(self, x):
 
  torch_model = Net() # type: ignore
  torch_model.train(False)
- _test_torch_model_single_io(torch_model, (3, 1, 256), (3, 1, 256), target_ios=13, decimal=1) # type: ignore
+ _test_torch_model_single_io(torch_model, (3, 1, 256), (3, 1, 256), target_ios='13', decimal=1) # type: ignore
 
 
  def test_1d_conv(self):