Address lint checker warning

gramalingam · web-flow · commit 5c181fdeeaa3 · 2025-11-09T20:03:29.000-08:00
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -9192,7 +9192,7 @@ def aten_unique_consecutive(
     dim: Optional[int] = None,
 ) -> tuple[TensorType, TensorType, TensorType]:
     """unique_consecutive(Tensor self, bool return_inverse=False, bool return_counts=False, int? dim=None) -> (Tensor, Tensor, Tensor)"""
-    assert x.dtype == INT64.dtype or x.dtype == INT32.dtype, (
+    assert x.dtype in {INT64.dtype, INT32.dtype}, (
         "unique_consecutive not implemented for other type than int32, int64"
     )
     rank_x = len(x.shape)

Original file line number	Diff line number	Diff line change
`@@ -9192,7 +9192,7 @@ def aten_unique_consecutive(`
`9192`	`9192`	`dim: Optional[int] = None,`
`9193`	`9193`	`) -> tuple[TensorType, TensorType, TensorType]:`
`9194`	`9194`	`"""unique_consecutive(Tensor self, bool return_inverse=False, bool return_counts=False, int? dim=None) -> (Tensor, Tensor, Tensor)"""`
`9195`		`- assert x.dtype == INT64.dtype or x.dtype == INT32.dtype, (`
	`9195`	`+ assert x.dtype in {INT64.dtype, INT32.dtype}, (`
`9196`	`9196`	`"unique_consecutive not implemented for other type than int32, int64"`
`9197`	`9197`	`)`
`9198`	`9198`	`rank_x = len(x.shape)`