pass the desired frame to traceback.walk_stack for 3.11

see python/cpython#96092
and it's not really intended for use outside of StackSummary.extract,
see python/cpython#99015 (comment)

Author: graingert

distributed/client.py

@@ -2910,7 +2910,7 @@ def run(
         )
 
     @staticmethod
-    def _get_computation_code(stacklevel=None) -> str:
+    def _get_computation_code(stacklevel: int | None = None) -> str:
         """Walk up the stack to the user code and extract the code surrounding
         the compute/submit/persist call. All modules encountered which are
         ignored through the option
@@ -2921,7 +2921,6 @@ def _get_computation_code(stacklevel=None) -> str:
         ``stacklevel`` may be used to explicitly indicate from which frame on
         the stack to get the source code.
         """
-
         ignore_modules = dask.config.get(
             "distributed.diagnostics.computations.ignore-modules"
         )
@@ -2939,7 +2938,8 @@ def _get_computation_code(stacklevel=None) -> str:
         else:
             # stacklevel 0 or less - shows dask internals which likely isn't helpful
             stacklevel = stacklevel if stacklevel > 0 else 1
-        for i, (fr, _) in enumerate(traceback.walk_stack(None), 1):
+
+        for i, (fr, _) in enumerate(traceback.walk_stack(sys._getframe().f_back), 1):
             if stacklevel is not None:
                 if i != stacklevel:
                     continue