Fix type hints for entrypoint decorator (#3198)

This commit is contained in:
Nuno Campos
2025-01-24 15:07:21 -08:00
committed by GitHub
+1 -2
View File
@@ -2,7 +2,6 @@ import asyncio
import concurrent.futures
import functools
import inspect
import types
from dataclasses import dataclass
from typing import (
Any,
@@ -402,7 +401,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: