Passing target_ios correctly to test_gru (#477)

bhushan23 · web-flow · commit f0a9327ce296 · 2019-09-17T11:55:49.000-07:00
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):
