Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
34 commits
Select commit Hold shift + click to select a range
6ae1aa2
Add hooks for task submission and resolution
cswartzvi Feb 24, 2025
13b8f98
Make `TaskImplementation` hashable by task ID
cswartzvi Feb 24, 2025
fda3f64
Add calls to task submission and resolution hooks
cswartzvi Feb 24, 2025
d6df6e4
Add tests for task submission and resolution hooks
cswartzvi Mar 2, 2025
9892959
Add docs for task submission and resolution hooks
cswartzvi Mar 2, 2025
2e31305
Update comment 'FIXME' -> 'TODO'
cswartzvi Mar 2, 2025
7e9b70b
Convert `str | None` to `Optional[str]`
cswartzvi Mar 2, 2025
4efee2c
Fix docstring typos
cswartzvi Mar 2, 2025
e5d0086
Merge branch 'DAGWorks-Inc:main' into task_submission_hook
cswartzvi Mar 3, 2025
a19de6e
Make `TaskFuture` a protocol, add `TaskFutureWrappingFunction`
cswartzvi Mar 15, 2025
e9b7a4e
Add better exception handling; clean up code
cswartzvi Mar 15, 2025
a8b042a
Add imports for task and submission hooks
cswartzvi Mar 15, 2025
627bf1b
Make sure finally block raises exception
cswartzvi Mar 17, 2025
c29eb1a
Add comments regarding task counts
cswartzvi Mar 17, 2025
d9754b8
Add submission and resolution hooks to `LifecycleAdapter` union
cswartzvi Mar 17, 2025
87e8743
Check cache in `TaskFutureWrappingFunction.get_state`
cswartzvi Mar 20, 2025
5af0ded
Merge branch 'DAGWorks-Inc:main' into task_submission_hook
cswartzvi Mar 22, 2025
a5ba413
Rename 'task resolution' to 'task return'
cswartzvi Mar 22, 2025
46c7f6a
Add context-aware logging adapters (plus helper functions)
cswartzvi Mar 17, 2025
e638ffb
Add tests for Add context-aware logging adapters
cswartzvi Mar 17, 2025
8504e06
Fix docstrings
cswartzvi Mar 17, 2025
9aaf948
Fix typo
cswartzvi Mar 17, 2025
11fb134
Fix typo
cswartzvi Mar 17, 2025
8492513
Fix typo
cswartzvi Mar 17, 2025
ad45dc7
Fix typo
cswartzvi Mar 17, 2025
370e4de
Add message about duplicate exception logs
cswartzvi Mar 17, 2025
d5ab9f3
Temporarily remove dask and ray test
cswartzvi Mar 17, 2025
9d09ef0
Make type hints backward compatible
cswartzvi Mar 20, 2025
a53a61f
Fix log typo 🤦‍♂️ and test assertion
cswartzvi Mar 20, 2025
ad22466
Add context check; move task nodes to separate file
cswartzvi Mar 20, 2025
37f36ea
Add default `extra` parameter for python <3.10
cswartzvi Mar 21, 2025
713538a
Use empty dict for default `extra` parameters
cswartzvi Mar 21, 2025
de99d49
Update 'task resolution' to 'task return'
cswartzvi Mar 22, 2025
8caab04
Merge branch 'main' into logging_adapter
elijahbenizzy Mar 29, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion hamilton/lifecycle/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -764,7 +764,7 @@ def post_task_expand(self, *, run_id: str, task_id: str, parameters: Dict[str, A

@abc.abstractmethod
def run_after_task_grouping(self, *, run_id: str, task_ids: List[str], **future_kwargs):
"""Runs after task grouping. This allows you to capture information about the which tasks were
"""Runs after task grouping. This allows you to capture information about which tasks were
created for a given run.

:param run_id: ID of the run, unique in scope of the driver.
Expand Down
Loading