Skip to content

Commit 8c59d0f

Browse files
committed
update prediction template
1 parent a398d47 commit 8c59d0f

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

argo-workflow/automlai-prediction-template.yaml

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,21 +3,21 @@ kind: ClusterWorkflowTemplate
33
metadata:
44
name: automlai-prediction
55
spec:
6-
entrypoint: run-amlp
6+
entrypoint: run-prediction
77
arguments:
88
parameters:
99
- name: url
10-
value: http://sunrise-mlflow-tracking.mlflow:5080
10+
value: http://mlflow.isiath.duckdns.org:8082
1111
- name: input
1212
value: iris_reg.csv
1313
- name: predictiontype
1414
value: regression
1515
- name: outputloc
1616
value: output.txt
1717
- name: runid
18-
value: NONE
18+
value: 064fb7a188d34a3da87f2271b8d8d9c2
1919
templates:
20-
- name: run-amlp
20+
- name: run-prediction
2121
inputs:
2222
parameters:
2323
- name: url
@@ -34,9 +34,9 @@ spec:
3434
image: ppalmes/automlai:v2.0
3535
command:
3636
args:
37-
- "-u{{inputs.parameters.url}}"
38-
- "-t{{inputs.parameters.predictiontype}}"
39-
- "-o/outputfile"
37+
- "-u {{inputs.parameters.url}}"
38+
- "-t {{inputs.parameters.predictiontype}}"
39+
- "-o /outputfile"
4040
- "--predict_only"
4141
- "--runid {{inputs.parameters.runid}}"
4242
- "/inputfile/{{inputs.parameters.input}}"
@@ -45,4 +45,4 @@ spec:
4545
- name: ofile
4646
path: /outputfile
4747
s3:
48-
key: "output/{{inputs.parameters.predictiontype}}_{{inputs.parameters.input}}.tgz"
48+
key: "output/{{inputs.parameters.predictiontype}}_{{inputs.parameters.input}}_{{inputs.parameters.outputloc}}.tgz"

0 commit comments

Comments
 (0)