Skip to content

Commit

Permalink
fix lint errors
Browse files Browse the repository at this point in the history
  • Loading branch information
Lunderberg committed Nov 7, 2023
1 parent 10dc5c9 commit 1fb2578
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions python/tvm/relax/op/tuple.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
from ..expr import Expr, PrimValue


def tuple_get_item(tuple: Expr, index: Union[int, PrimExpr, Expr]) -> Expr:
def tuple_get_item(tuple_expr: Expr, index: Union[int, PrimExpr, Expr]) -> Expr:
"""Perform tuple access
Use of this method is recommended, rather than constructing a
Expand All @@ -38,7 +38,7 @@ def tuple_get_item(tuple: Expr, index: Union[int, PrimExpr, Expr]) -> Expr:
Parameters
----------
tuple: Expr
tuple_expr: Expr
The tuple to be accessed. The tuple is not required to be an
in-line `relax.Tuple`, but must have `TupleStructInfo`
Expand All @@ -58,10 +58,10 @@ def tuple_get_item(tuple: Expr, index: Union[int, PrimExpr, Expr]) -> Expr:
if not isinstance(index, Expr):
index = PrimValue(index)

return _ffi_api.tuple_get_item(tuple, index) # type: ignore
return _ffi_api.tuple_get_item(tuple_expr, index) # type: ignore


def tuple_get_item_dyn(tuple: Expr, index: Union[int, PrimExpr, Expr]) -> Expr:
def tuple_get_item_dyn(tuple_expr: Expr, index: Union[int, PrimExpr, Expr]) -> Expr:
"""Explicitly generate a call to tuple_get_item_dyn
This method is not recommended for general use, and is provided to
Expand All @@ -71,7 +71,7 @@ def tuple_get_item_dyn(tuple: Expr, index: Union[int, PrimExpr, Expr]) -> Expr:
Parameters
----------
tuple: Expr
tuple_expr: Expr
The tuple to be accessed. The tuple is not required to be an
in-line `relax.Tuple`, but must have `TupleStructInfo`
Expand All @@ -90,4 +90,4 @@ def tuple_get_item_dyn(tuple: Expr, index: Union[int, PrimExpr, Expr]) -> Expr:
"""
if not isinstance(index, Expr):
index = PrimValue(index)
return tvm.relax.Call(tvm.ir.Op.get("relax.tuple_get_item_dyn"), [tuple, index])
return tvm.relax.Call(tvm.ir.Op.get("relax.tuple_get_item_dyn"), [tuple_expr, index])

0 comments on commit 1fb2578

Please sign in to comment.