revert auto_heartbeater annotation change

Author: jssmith

custom_decorator/activity_utils.py

@@ -8,7 +8,7 @@
 F = TypeVar("F", bound=Callable[..., Awaitable[Any]])
 
 
-def _auto_heartbeater(fn: F) -> F:
+def auto_heartbeater(fn: F) -> F:
     # We want to ensure that the type hints from the original callable are
     # available via our wrapper, so we use the functools wraps decorator
     @wraps(fn)

custom_decorator/worker.py

@@ -6,13 +6,13 @@
 from temporalio.common import RetryPolicy
 from temporalio.worker import Worker
 
-from custom_decorator.activity_utils import _auto_heartbeater
+from custom_decorator.activity_utils import auto_heartbeater
 
 
 # Here we use our automatic heartbeater decorator. If this wasn't present, our
 # activity would timeout since it isn't heartbeating.
 @activity.defn
-@_auto_heartbeater
+@auto_heartbeater
 async def wait_for_cancel_activity() -> str:
     # Wait forever, catch the cancel, and return some value
     try: