Fix gpu limit issue

Wh1isper · Jul 19, 2024 · 4b125e5 · 4b125e5
1 parent 22be5aa
commit 4b125e5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/moriarty/matrix/operator_/spawner/impl/k8s.py b/moriarty/matrix/operator_/spawner/impl/k8s.py
@@ -331,6 +331,7 @@ def _make_compute_contaienr(
  requests={
  "cpu": f"{endpoint_orm.cpu_request}",
  "memory": f"{endpoint_orm.memory_request}Mi",
+ **gpu_limit,
  },
  limits={
  **cpu_limit,