From f0a9327ce296c49720ae0d3bf5b5d81aa615db41 Mon Sep 17 00:00:00 2001 From: Bhushan Sonawane Date: Tue, 17 Sep 2019 11:55:49 -0700 Subject: [PATCH] Passing target_ios correctly to `test_gru` (#477) --- tests/pytorch_model_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pytorch_model_test.py b/tests/pytorch_model_test.py index 2fafab1..d925bfd 100644 --- 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):