Skip to content

Commit

Permalink
Update function name; revert bugfix
Browse files Browse the repository at this point in the history
  • Loading branch information
cloudw committed Aug 19, 2024
1 parent cc5c597 commit ee92317
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 5 deletions.
5 changes: 3 additions & 2 deletions metaflow/plugins/aip/aip.py
Original file line number Diff line number Diff line change
Expand Up @@ -291,9 +291,10 @@ def _create_workflow_yaml(
# Note the name has to follow k8s format.
# self.name is typically CamelCase as it's python class name.
# generateName contains a sanitized version of self.name from aip.compiler
default_workflow_name = workflow["metadata"].pop("generateName").rstrip("-")
workflow["metadata"]["name"] = (
sanitize_k8s_name(name) if name else default_workflow_name
sanitize_k8s_name(name)
if name
else workflow["metadata"].pop("generateName").rstrip("-")
)

# Service account is added through webhooks.
Expand Down
4 changes: 2 additions & 2 deletions metaflow/plugins/aip/argo_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def __init__(self, kubernetes_namespace: str = KUBERNETES_NAMESPACE):
"""
self._client = ArgoClient(namespace=kubernetes_namespace)

def trigger(
def trigger_exact(
self,
template_name: Optional[str] = None,
parameters: Optional[dict] = None,
Expand Down Expand Up @@ -100,7 +100,7 @@ def trigger_latest(
filter_func=filter_func,
)

return self.trigger(
return self.trigger_exact(
template_name=template_name,
parameters=parameters,
wait_timeout=wait_timeout,
Expand Down
2 changes: 1 addition & 1 deletion metaflow/plugins/aip/tests/flows/flow_triggering_flow.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,7 @@ def end(self):

# ====== Test template triggering ======
logger.info("\n Testing ArgoHelper.trigger")
run_id, run_uid = argo_helper.trigger(
run_id, run_uid = argo_helper.trigger_exact(
template_name=self.workflow_template_names[0],
parameters={
"trigger_enabled": False,
Expand Down

0 comments on commit ee92317

Please sign in to comment.