nit fixes

Author: akhilg-nv

tripy/nvtripy/frontend/ops/_registry.py

@@ -16,7 +16,6 @@
 #
 
 from typing import Any, Callable
-from functools import wraps
 
 # We use the tensor method registry to define methods on the `Tensor` class out of line.
 # This lets the method live alongside the trace operation and makes it a bit more modular

tripy/nvtripy/utils/ast.py

@@ -123,6 +123,7 @@ def get_arg_candidate_column_offsets(
                 arg_node = node.keywords[index - num_positional]
             else:
                 # Detect method calls by examining AST structure
+                # For methods, the `self` argument is omitted from ast.Call.args
                 is_method_call = isinstance(node.func, ast.Attribute)
                 if is_method_call:
                     index -= 1