Skip to content

Commit

Permalink
add additional field to agent create with task execution metadata
Browse files Browse the repository at this point in the history
Signed-off-by: noahjax <[email protected]>
  • Loading branch information
noahjax committed Mar 26, 2024
1 parent a1af346 commit e81c592
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions tests/flytekit/unit/extend/test_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,10 +55,7 @@
class DummyMetadata(ResourceMeta):
job_id: str
output_path: typing.Optional[str] = None
<<<<<<< HEAD
task_name: typing.Optional[str] = None
=======
>>>>>>> cdf11c88 (add simple test to check that AysncAgent subclass can consume output_prefix)


class DummyAgent(AsyncAgentBase):
Expand Down Expand Up @@ -198,7 +195,6 @@ async def test_async_agent_service(agent, consume_metadata):

inputs_proto = task_inputs.to_flyte_idl()
output_prefix = "/tmp"
<<<<<<< HEAD
metadata_bytes = (
DummyMetadata(
job_id=dummy_id,
Expand All @@ -208,9 +204,6 @@ async def test_async_agent_service(agent, consume_metadata):
if consume_metadata
else DummyMetadata(job_id=dummy_id).encode()
)
=======
metadata_bytes = DummyMetadata(job_id=dummy_id, output_path=f"{output_prefix/{dummy_id}}").encode()
>>>>>>> cdf11c88 (add simple test to check that AysncAgent subclass can consume output_prefix)

tmp = get_task_template(agent.task_category.name).to_flyte_idl()
task_category = TaskCategory(name=agent.task_category.name, version=0)
Expand Down

0 comments on commit e81c592

Please sign in to comment.