File tree Expand file tree Collapse file tree 2 files changed +2
-5
lines changed
tests/tests_pytorch/models Expand file tree Collapse file tree 2 files changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -180,6 +180,7 @@ markers = [
180180]
181181filterwarnings = [
182182 " error::FutureWarning" ,
183+ " ignore::FutureWarning:onnxscript" , # Temporary ignore until onnxscript is updated
183184]
184185xfail_strict = true
185186junit_duration_report = " call"
Original file line number Diff line number Diff line change 1313# limitations under the License.
1414import operator
1515import os
16- import warnings
1716from io import BytesIO
1817from pathlib import Path
1918from unittest .mock import patch
@@ -176,8 +175,5 @@ def test_model_return_type():
176175 model .example_input_array = torch .randn ((1 , 32 ))
177176 model .eval ()
178177
179- # Temporarily suppress FutureWarning from onnxscript internal function.
180- with warnings .catch_warnings ():
181- warnings .simplefilter ("ignore" , FutureWarning )
182- ret = model .to_onnx (dynamo = True )
178+ ret = model .to_onnx (dynamo = True )
183179 assert isinstance (ret , torch .onnx .ONNXProgram )
You can’t perform that action at this time.
0 commit comments