lint

Author: justinchuby

onnxscript/function_libs/torch_aten/ops/core.py

@@ -3453,7 +3453,9 @@ def aten_new_empty_strided(self: TensorType, size: INT64, stride: INT64) -> Tens
 
 
 @torch_op("aten::new_full")
-def aten_new_full(self, size: INT64, fill_value, dtype: int = FLOAT.dtype):
+def aten_new_full(
+    self, size: INT64, fill_value, dtype: int = FLOAT.dtype
+):  # pylint: disable=unused-argument
     # new_full(Tensor self, SymInt[] size, Scalar fill_value, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
 
     fill_value = op.Cast(fill_value, to=dtype)

onnxscript/test/function_libs/torch_aten/ops_correctness_test.py

@@ -169,6 +169,7 @@ def wrapped(fn):
     "exp": core_ops.aten_exp,
     "exp2": core_ops.aten_exp2,
     "fmod": core_ops.aten_fmod,
+    # TODO(justinchuby): Test aten::full
     "full_like": core_ops.aten_full_like,
     "gt": core_ops.aten_gt,
     "isinf": core_ops.aten_isinf,