Fix the infra_workload error

Fix the yaml loader error which results in failure to load
VF modules

Change-Id: I1d9a699b9fffd8c9c84e20a8cc9aaa4762c2e7f3
Issue-ID: MULTICLOUD-656
Signed-off-by: Bin Yang <bin.yang@windriver.com>

diff --git a/share/starlingx_base/resource/infra_workload.py b/share/starlingx_base/resource/infra_workload.py
index 48990d2..9524d67 100644 (file)
--- a/share/starlingx_base/resource/infra_workload.py
+++ b/share/starlingx_base/resource/infra_workload.py
@@ -467,7 +467,7 @@ class InfraWorkloadHelper(infra_workload_helper.InfraWorkloadHelper):
                                               'r') as af:
                                         # assume the template file size is small
                                         templatedata1["template"] = \
-                                            yaml.safe_load(af.read(), Loader=NoDatesSafeLoader)
+                                            yaml.load(af, Loader=NoDatesSafeLoader)
                                     # pass
 
                                 elif artifact_type.lower() == "heat_env":
@@ -475,8 +475,8 @@ class InfraWorkloadHelper(infra_workload_helper.InfraWorkloadHelper):
                                     with open(artifact_path,
                                               'r') as af:
                                         # assume the env file size is small
-                                        templatedata1.update(yaml.safe_load(
-                                            af.read(), Loader=NoDatesSafeLoader))
+                                        templatedata1.update(yaml.load(
+                                            af, Loader=NoDatesSafeLoader))
                                     # pass
                                 # pass
                             return templatedata1