-
Notifications
You must be signed in to change notification settings - Fork 3
/
project.yaml
144 lines (144 loc) · 4.05 KB
/
project.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
kind: project
metadata:
name: ny-demo-lgbm-model-1
created: '2023-02-16T15:06:21.174000'
spec:
functions:
- name: data-prep
kind: job
image: mlrun/ml-models
handler: src.data_prep.data_preparation
with_repo: true
- name: trainer
kind: job
image: mlrun/ml-models
handler: src.trainer.train
with_repo: true
- url: hub://auto_trainer
name: evaluate
kind: job
image: mlrun/ml-models
handler: evaluate
- name: serving
spec:
kind: serving
metadata:
name: serving
project: ny-demo-lgbm-mode-1
categories: []
spec:
command: ''
args: []
image: mlrun/ml-models
build:
source: ./
commands: []
code_origin: git://github.com/davesh0812/nyc-taxi-demo.git#refs/heads/main#f21c030122144c4de8588718bbb435434eadbf67
description: ''
disable_auto_mount: false
env: []
resources:
requests:
memory: 1Mi
cpu: 25m
limits:
memory: 20Gi
cpu: '2'
priority_class_name: igz-workload-medium
preemption_mode: prevent
min_replicas: 1
max_replicas: 4
source: ''
function_kind: serving_v2
base_image_pull: false
graph:
steps:
calculate_airport_distance:
kind: task
handler: src.data_prep.add_airport_dist
after: []
calculate_radian_conv:
kind: task
handler: src.data_prep.radian_conv_step
after:
- calculate_airport_distance
bearing_calculation:
kind: task
handler: src.data_prep.sphere_dist_bear_step
after:
- calculate_radian_conv
distance_calculation:
kind: task
handler: src.data_prep.sphere_dist_step
after:
- bearing_calculation
DateExtractor:
kind: task
class_name: mlrun.feature_store.steps.DateExtractor
class_args:
parts:
- hour
- day
- month
- day_of_week
- year
timestamp_col: pickup_datetime
after:
- distance_calculation
preprocess:
kind: task
handler: src.serving.preprocess
after:
- DateExtractor
predict_fare:
kind: task
class_name: mlrun.frameworks.lgbm.LGBMModelServer
class_args:
model_path: ''
after:
- preprocess
postprocess:
kind: task
handler: src.serving.postprocess
after:
- predict_fare
responder: true
engine: async
secret_sources: []
affinity:
nodeAffinity:
requiredDuringSchedulingIgnoredDuringExecution:
nodeSelectorTerms:
- matchExpressions:
- key: app.iguazio.com/lifecycle
operator: NotIn
values:
- preemptible
- key: eks.amazonaws.com/capacityType
operator: NotIn
values:
- SPOT
- key: node-lifecycle
operator: NotIn
values:
- spot
tolerations: null
security_context: {}
verbose: false
- name: server_tester
kind: job
image: mlrun/ml-models
handler: src.serving_test.model_server_tester
with_repo: true
workflows:
- path: src/train_workflow.py
name: train_workflow
- path: src/predict_workflow.py
name: predict_workflow
artifacts: []
source: git://github.com/davesh0812/nyc-taxi-demo.git#refs/heads/main
origin_url: git://github.com/davesh0812/nyc-taxi-demo.git#refs/heads/main
load_source_on_run: true
desired_state: online
status:
state: online