Skip to content

Commit

Permalink
Agent Get Pending Message (flyteorg#1834)
Browse files Browse the repository at this point in the history
Signed-off-by: Future Outlier <[email protected]>
Signed-off-by: Kevin Su <[email protected]>
Co-authored-by: Future Outlier <[email protected]>
Co-authored-by: Kevin Su <[email protected]>
Co-authored-by: Kevin Su <[email protected]>
Signed-off-by: Rafael Raposo <[email protected]>
  • Loading branch information
4 people authored and RRap0so committed Dec 15, 2023
1 parent fc86ee9 commit 774f857
Show file tree
Hide file tree
Showing 6 changed files with 224 additions and 96 deletions.
2 changes: 1 addition & 1 deletion flytekit/extend/backend/base_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -215,7 +215,7 @@ async def _get(self, resource_meta: bytes) -> GetTaskResponse:
else:
res = self._agent.get(grpc_ctx, resource_meta)
state = res.resource.state
logger.info(f"Task state: {state}")
logger.info(f"Task state: {state}, State message: {res.resource.message}")
return res

def signal_handler(self, resource_meta: bytes, signum: int, frame: FrameType) -> typing.Any:
Expand Down
2 changes: 1 addition & 1 deletion plugins/flytekit-papermill/dev-requirements.in
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
flyteidl>=1.3.0
flyteidl>=1.10.0
-e file:../../.#egg=flytekitplugins-pod&subdirectory=plugins/flytekit-k8s-pod
-e file:../../.#egg=flytekitplugins-spark&subdirectory=plugins/flytekit-spark
-e file:../../.#egg=flytekitplugins-awsbatch&subdirectory=plugins/flytekit-aws-batch
Loading

0 comments on commit 774f857

Please sign in to comment.