SAASMLOPS-1403 Add tolerations to bytewax pod spec (#18)

sailpoint · Jul 23, 2024 · a5030cb · a5030cb
1 parent e980f33
commit a5030cb
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/sdk/python/feast/infra/materialization/kubernetes/k8s_materialization_engine.py b/sdk/python/feast/infra/materialization/kubernetes/k8s_materialization_engine.py
@@ -416,6 +416,14 @@ def _create_job_definition(self, job_id, namespace, pods, env, index_offset=0):
                                 "name": self._configmap_name(job_id),
                             },
                         ],
+                        "tolerations": [
+                            {
+                                "key": "bytewax",
+                                "operator": "Equal",
+                                "value": "instance",
+                                "effect": "NoSchedule",
+                            }
+                        ],
                     },
                 },
             },