Skip to content

Commit 9fd572f

Browse files
authored
Kubernetes: fix is_hard_taint check (#3803)
Fixes: #3802
1 parent aca433d commit 9fd572f

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

src/dstack/_internal/core/backends/kubernetes/resources.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -207,13 +207,14 @@ def get_node_labels(node: V1Node) -> dict[str, str]:
207207

208208

209209
def is_hard_taint(taint: V1Taint) -> bool:
210-
if taint.effect == TaintEffect.PREFER_NO_SCHEDULE:
211-
return False
212-
if taint.effect not in TaintEffect:
210+
try:
211+
taint_effect = TaintEffect(taint.effect)
212+
except ValueError:
213213
logger.warning(
214214
"Unexpected taint %s=%s effect: %s", taint.key, taint.value or "", taint.effect
215215
)
216-
return True
216+
return True
217+
return taint_effect is not TaintEffect.PREFER_NO_SCHEDULE
217218

218219

219220
def is_taint_tolerated(taint: V1Taint) -> bool:

0 commit comments

Comments
 (0)