fix formatting

Signed-off-by: Achille Roussel <achille.roussel@gmail.com>

Author: achille-roussel

src/dispatch/experimental/durable/function.py

@@ -137,7 +137,9 @@ def __getstate__(self):
             if frame_state < FRAME_CLEARED:
                 print(f"IP = {ip}")
                 print(f"SP = {sp}")
-                for i, (is_null, value) in enumerate(stack if stack is not None else []):
+                for i, (is_null, value) in enumerate(
+                    stack if stack is not None else []
+                ):
                     if is_null:
                         print(f"stack[{i}] = NULL")
                     else:

src/dispatch/experimental/durable/registry.py

@@ -42,7 +42,7 @@ def __setstate__(self, state):
 
         # mypy 1.10.0 seems to report a false positive here:
         # error: Incompatible types in assignment (expression has type "FunctionType", variable has type "MethodType")  [assignment]
-        self.fn = rfn.fn # type: ignore
+        self.fn = rfn.fn  # type: ignore
         self.key = key
         self.filename = filename
         self.lineno = lineno

src/dispatch/function.py

@@ -258,7 +258,9 @@ def primitive_func(input: Input) -> Output:
         primitive_func.__qualname__ = f"{name}_primitive"
         durable_primitive_func = durable(primitive_func)
 
-        wrapped_func = Function[P, T](self.endpoint, self.client, name, durable_primitive_func)
+        wrapped_func = Function[P, T](
+            self.endpoint, self.client, name, durable_primitive_func
+        )
         self._register(name, wrapped_func)
         return wrapped_func
 

tests/dispatch/test_scheduler.py

@@ -1,5 +1,5 @@
 import unittest
-from typing import Any, Callable, List, Optional, Type, Set
+from typing import Any, Callable, List, Optional, Set, Type
 
 from dispatch.coroutine import AnyException, any, call, gather, race
 from dispatch.experimental.durable import durable