From d347f60bf058f847094b11d4ee862046deaa8ce0 Mon Sep 17 00:00:00 2001 From: oldhu Date: Tue, 1 Jul 2025 23:36:04 +0800 Subject: [PATCH 1/2] fix Application.arun's returning type to include [ApplicationStateType] Align with Application.run and fix the type checker issue on calling arun --- burr/core/application.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/burr/core/application.py b/burr/core/application.py index c77ba9302..080b090f7 100644 --- a/burr/core/application.py +++ b/burr/core/application.py @@ -1297,7 +1297,7 @@ async def arun( halt_before: list[str] = None, halt_after: list[str] = None, inputs: Optional[Dict[str, Any]] = None, - ) -> Tuple[Action, Optional[dict], State]: + ) -> Tuple[Action, Optional[dict], State[ApplicationStateType]]: """Runs your application through until completion, using async. Does not give access to the state along the way -- if you want that, use iterate(). From 84d62116f7d6aa5e84cf6d1ffda007f8470f26f2 Mon Sep 17 00:00:00 2001 From: oldhu Date: Wed, 2 Jul 2025 14:44:04 +0800 Subject: [PATCH 2/2] add typing information for SyncTrackingClient clone --- burr/tracking/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/burr/tracking/base.py b/burr/tracking/base.py index f5813e2f1..d55208b56 100644 --- a/burr/tracking/base.py +++ b/burr/tracking/base.py @@ -31,7 +31,7 @@ class SyncTrackingClient( TODO -- create an async tracking client""" @abc.abstractmethod - def copy(self): + def copy(self) -> "SyncTrackingClient": pass