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

sailpoint · Jul 24, 2024 · 5bd03bc · 5bd03bc
1 parent 1673e19
commit 5bd03bc
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
@@ -414,6 +414,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",
+                            }
+                        ],
                     },
                 },
             },