diff --git a/onnxscript/function_libs/torch_lib/ops/nn.py b/onnxscript/function_libs/torch_lib/ops/nn.py index 022f4c317..7cb069cab 100644 --- a/onnxscript/function_libs/torch_lib/ops/nn.py +++ b/onnxscript/function_libs/torch_lib/ops/nn.py @@ -2417,7 +2417,7 @@ def aten_upsample_bicubic2d( ) -@torch_op(("aten::upsample_bicubic2d.vec", "aten::upsample_linear1d.vec", trace_only=True) +@torch_op(("aten::upsample_bicubic2d.vec", "aten::upsample_linear1d.vec"), trace_only=True) def aten_upsample_bicubic2d_vec( self: TReal, output_size: INT64,