use traceback.extract_stack(limit=2)[0]

Looks walk_stack() use an incorrect current frame

python/cudf/cudf/utils/utils.py

@@ -159,14 +159,9 @@ def _external_only_api(func, alternative=""):
     @functools.wraps(func)
     def wrapper(*args, **kwargs):
         # Check the immediately preceding frame to see if it's in cudf.
-        tb = traceback.walk_stack(None)
-        frame, lineno = next(tb)
-        fn = frame.f_code.co_filename
-        if "performance_tracking.py" in fn:
-            # Go one frame further back if the caller is wrapper in a
-            # _performance_tracking decorator
-            frame, lineno = next(tb)
-            fn = frame.f_code.co_filename
+        pre_frame = traceback.extract_stack(limit=2)[0]
+        fn = pre_frame.filename
+        lineno = pre_frame.lineno
         if _cudf_root in fn and _tests_root not in fn:
             raise RuntimeError(
                 f"External-only API called in {fn} at line {lineno}. "