Fix lint error

stack-info: PR: #926, branch: jansel/stack/187

Author: jansel

helion/_compiler/device_function.py

@@ -509,23 +509,25 @@ def _format_constexpr_value(self, value: object) -> str:
         # Extract sympy expression from torch symbolic types
         if isinstance(value, (torch.SymInt, torch.SymFloat, torch.SymBool)):
             value = value._sympy_()
+            if isinstance(value, tuple):
+                # torch symbolic values may return (expr, replacements); use the expr
+                value = value[0]
 
         # Handle sympy expressions (sanitize by replacing triton_helpers functions)
         if isinstance(value, sympy.Expr):
-            expr = cast(
-                "sympy.Expr",
-                value.replace(
-                    lambda node: isinstance(node, sympy.Function)
-                    and getattr(node.func, "__name__", "")
-                    == "triton_helpers.div_floor_integer",
-                    lambda node: sympy.floor(node.args[0] / node.args[1]),  # pyright: ignore[reportAttributeAccessIssue]
-                ).replace(
-                    lambda node: isinstance(node, sympy.Function)
-                    and getattr(node.func, "__name__", "")
-                    == "triton_helpers.remainder_integer",
-                    lambda node: sympy.Mod(node.args[0], node.args[1]),  # pyright: ignore[reportAttributeAccessIssue]
-                ),
+            sympy_expr = cast("sympy.Expr", value)
+            sanitized = sympy_expr.replace(  # pyright: ignore[reportAttributeAccessIssue]
+                lambda node: isinstance(node, sympy.Function)
+                and getattr(node.func, "__name__", "")
+                == "triton_helpers.div_floor_integer",
+                lambda node: sympy.floor(node.args[0] / node.args[1]),  # pyright: ignore[reportAttributeAccessIssue]
+            ).replace(  # pyright: ignore[reportAttributeAccessIssue]
+                lambda node: isinstance(node, sympy.Function)
+                and getattr(node.func, "__name__", "")
+                == "triton_helpers.remainder_integer",
+                lambda node: sympy.Mod(node.args[0], node.args[1]),  # pyright: ignore[reportAttributeAccessIssue]
             )
+            expr = cast("sympy.Expr", sanitized)
             return HostFunction.current().sympy_expr(expr)
 
         return HostFunction.current().literal_expr(value)