chore: yaml safe load

Signed-off-by: ThibaultFy <[email protected]>
Substra · Apr 25, 2024 · 0df805e · 0df805e
1 parent bd89e78
commit 0df805e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/backend/substrapp/kubernetes_utils.py b/backend/substrapp/kubernetes_utils.py
@@ -52,7 +52,9 @@ def get_resources_requirements_from_yaml(
     *,
     yaml_resources: str,
 ) -> kubernetes.client.V1ResourceRequirements:
-    resources_dict = yaml.load(yaml_resources, Loader=yaml.FullLoader)
+    """Return a kubernetes.client.V1ResourceRequirements object from a yaml string."""
+    resources_dict = yaml.safe_load(yaml_resources)
+
     return kubernetes.client.V1ResourceRequirements(
         requests=resources_dict["requests"], limits=resources_dict["limits"]
     )