Skip to content

Commit 6a5c14d

Browse files
committed
Update runtime base class to add defined tags.
1 parent 13b7ba4 commit 6a5c14d

File tree

1 file changed

+21
-4
lines changed

1 file changed

+21
-4
lines changed

ads/jobs/builders/runtimes/base.py

Lines changed: 21 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,12 @@ class Runtime(Builder):
2424
CONST_ENV_VAR = "env"
2525
CONST_ARGS = "args"
2626
CONST_MAXIMUM_RUNTIME_IN_MINUTES = "maximumRuntimeInMinutes"
27-
CONST_TAG = "freeformTags"
27+
CONST_FREEFORM_TAGS = "freeformTags"
28+
CONST_DEFINED_TAGS = "definedTags"
2829

2930
attribute_map = {
30-
CONST_TAG: "freeform_tags",
31+
CONST_FREEFORM_TAGS: "freeform_tags",
32+
CONST_DEFINED_TAGS: "defined_tags",
3133
CONST_ENV_VAR: CONST_ENV_VAR,
3234
}
3335

@@ -171,7 +173,17 @@ def with_freeform_tag(self: Self, **kwargs) -> Self:
171173
Self
172174
This method returns self to support chaining methods.
173175
"""
174-
return self.set_spec(self.CONST_TAG, kwargs)
176+
return self.set_spec(self.CONST_FREEFORM_TAGS, kwargs)
177+
178+
def with_defined_tag(self: Self, **kwargs) -> Self:
179+
"""Sets defined tag
180+
181+
Returns
182+
-------
183+
Self
184+
This method returns self to support chaining methods.
185+
"""
186+
return self.set_spec(self.CONST_DEFINED_TAGS, kwargs)
175187

176188
def with_maximum_runtime_in_minutes(
177189
self: Self, maximum_runtime_in_minutes: int
@@ -210,7 +222,12 @@ def envs(self) -> dict:
210222
@property
211223
def freeform_tags(self) -> dict:
212224
"""freeform_tags"""
213-
return self.get_spec(self.CONST_TAG, {})
225+
return self.get_spec(self.CONST_FREEFORM_TAGS, {})
226+
227+
@property
228+
def defined_tags(self) -> dict:
229+
"""defined_tags"""
230+
return self.get_spec(self.CONST_DEFINED_TAGS, {})
214231

215232
@property
216233
def args(self) -> list:

0 commit comments

Comments
 (0)