chore: review guilhem

Signed-off-by: ThibaultFy <[email protected]>
Substra · May 16, 2024 · dc145b9 · dc145b9
1 parent e9325ed
commit dc145b9
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/backend/substrapp/kubernetes_utils.py b/backend/substrapp/kubernetes_utils.py
@@ -55,15 +55,9 @@ def get_resources_requirements_from_yaml(
     """Return a kubernetes.client.V1ResourceRequirements object from a yaml string."""
     resources_dict = yaml.safe_load(yaml_resources)
 
-    try:
-        requests = resources_dict["requests"]
-    except KeyError:
-        requests = None
+    requests = resources_dict.get("requests", None)
 
-    try:
-        limits = resources_dict["limits"]
-    except KeyError:
-        limits = None
+    limits = resources_dict.get("limits", None)
 
     return kubernetes.client.V1ResourceRequirements(requests=requests, limits=limits)