Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fast register for dynamic tasks #437

Merged
merged 13 commits into from
Apr 13, 2021
Merged

Fast register for dynamic tasks #437

merged 13 commits into from
Apr 13, 2021

Conversation

katrogan
Copy link
Contributor

Type

  • Bug Fix
  • Feature
  • Plugin

Are all requirements met?

  • Code completed
  • Smoke tested
  • Unit tests added
  • Code documentation added
  • Any pending items have an associated Issue

Complete description

Passes in fast register arguments and substitutes them at dynamic task compilation time.

Tracking Issue

flyteorg/flyte#824

Follow-up issue

NA

katrogan added 11 commits March 26, 2021 15:34
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
Signed-off-by: Katrina Rogan <[email protected]>
@codecov
Copy link

codecov bot commented Mar 29, 2021

Codecov Report

Merging #437 (258a54c) into master (12cae6a) will increase coverage by 0.01%.
The diff coverage is 92.59%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master     #437      +/-   ##
==========================================
+ Coverage   84.25%   84.27%   +0.01%     
==========================================
  Files         338      338              
  Lines       25528    25577      +49     
  Branches     2095     2104       +9     
==========================================
+ Hits        21509    21555      +46     
- Misses       3449     3451       +2     
- Partials      570      571       +1     
Impacted Files Coverage Δ
flytekit/common/translator.py 90.08% <ø> (ø)
...sts/flytekit/unit/core/test_dynamic_conditional.py 58.82% <ø> (ø)
flytekit/core/python_function_task.py 88.18% <90.00%> (+0.13%) ⬆️
tests/flytekit/unit/core/test_type_hints.py 95.40% <91.30%> (-0.15%) ⬇️
flytekit/core/context_manager.py 91.87% <100.00%> (+0.47%) ⬆️
plugins/tests/pod/test_pod.py 95.45% <100.00%> (ø)
tests/flytekit/unit/core/test_context_manager.py 100.00% <100.00%> (ø)

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 12cae6a...258a54c. Read the comment docs.

Signed-off-by: Katrina Rogan <[email protected]>
Copy link
Contributor

@wild-endeavor wild-endeavor left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

just a question for now, still thinking about stuff, sorry

with ctx.new_execution_context(
mode=ExecutionState.Mode.TASK_EXECUTION,
additional_context={
"dynamic_addl_distro": "s3::/my-s3-bucket/fast/123",
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
"dynamic_addl_distro": "s3::/my-s3-bucket/fast/123",
"dynamic_addl_distro": "s3://my-s3-bucket/fast/123",

"distribution could be retrieved"
)
logger.warn(f"ctx.execution_state.additional_context {ctx.execution_state.additional_context}")
sanitized_tasks = set()
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

isn't tasks already a set? why do we need another set?

@wild-endeavor wild-endeavor merged commit bbdfad4 into master Apr 13, 2021
max-hoffman pushed a commit to dolthub/flytekit that referenced this pull request Apr 29, 2021
max-hoffman pushed a commit to dolthub/flytekit that referenced this pull request May 11, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants