Fix type hints for entrypoint decorator

2026-09-13 05:07:51 +02:00 · 2025-01-24 09:31:27 -08:00
parent 996b120613
commit c04802a344
1 changed files with 1 additions and 1 deletions
@@ -309,7 +309,7 @@ class entrypoint:
        A value will always be saved even if it is None.
        """

-    def __call__(self, func: types.FunctionType) -> Pregel:
+    def __call__(self, func: Callable[..., Any]) -> Pregel:
        """Convert a function into a Pregel graph.

        Args: