-
Notifications
You must be signed in to change notification settings - Fork 459
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
Sagemaker add pipeline tags #3080
Conversation
@bcdurak i have not tested this, so if u can test it id be grateful |
LLM Finetuning template updates in |
E2E template updates in |
src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py
Outdated
Show resolved
Hide resolved
Classification template updates in |
@@ -332,6 +332,7 @@ def prepare_or_run_pipeline( | |||
else None | |||
), | |||
) | |||
pipeline_tags = step_settings.pipeline_tags |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ah another thing: This is taking the pipeline tags from the settings specified for a specific step? I'm guessing this is in a for-loop and will just use the settings of the latest step.
Instead, I think we should use the tags defined on the pipeline settings (self.get_settings(deployment)
)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
pfft right again. Im rusty AF. Sorry
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🦭
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Tested this on our AWS stack, everything is working as intended.
Describe changes
I added tags to sagemaker pipelines
Pre-requisites
Please ensure you have done the following:
develop
and the open PR is targetingdevelop
. If your branch wasn't based on develop read Contribution guide on rebasing branch to develop.Types of changes