Fix argument names in scan and while_loop wrappers

Updated the scan function to use correct argument names ('f' and 'pbar') for brainstate.transform.scan. Fixed while_loop to properly wrap cond_fun in a lambda, matching the expected signature.

Author: Routhleck

brainpy/_src/math/object_transform/controls.py

@@ -308,12 +308,12 @@ def scan(
       The stacked outputs of ``body_fun`` when scanned over the leading axis of the inputs.
     """
     return brainstate.transform.scan(
-        body_fun=body_fun,
+        f=body_fun,
         init=init,
         xs=operands,
         reverse=reverse,
         unroll=unroll,
-        progress_bar=brainstate.transform.ProgressBar() if progress_bar else None,
+        pbar=brainstate.transform.ProgressBar() if progress_bar else None,
     )
 
 
@@ -381,4 +381,4 @@ def while_loop(
     """
     if not isinstance(operands, (tuple, list)):
         operands = (operands,)
-    return brainstate.transform.while_loop(lambda x: body_fun(*x), cond_fun(*x), operands)
+    return brainstate.transform.while_loop(lambda x: body_fun(*x), lambda x: cond_fun(*x), operands)