updating yaml load to safe load

datmo · Nov 7, 2018 · 8cefb13 · 8cefb13
1 parent 571a229
commit 8cefb13
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/datmo/core/controller/deploy/deploy.py b/datmo/core/controller/deploy/deploy.py
@@ -177,7 +177,7 @@ def model_deploy(self, cluster_name):
             if datmo_deploy_config_path:
                 with open(datmo_deploy_config_path, 'r') as stream:
                     try:
-                        datmo_deploy = yaml.load(stream)
+                        datmo_deploy = yaml.safe_load(stream)
                         if datmo_deploy is not None:
                             files_exclude = datmo_deploy['deploy'][
                                 'files_exclude']

diff --git a/datmo/core/controller/deploy/driver/datmo_microservice.py b/datmo/core/controller/deploy/driver/datmo_microservice.py
@@ -46,7 +46,7 @@ def validate_deploy(self, home):
         if datmo_deploy_config_path:
             with open(datmo_deploy_config_path, 'r') as stream:
                 try:
-                    datmo_deploy = yaml.load(stream)
+                    datmo_deploy = yaml.safe_load(stream)
                     if datmo_deploy is not None:
                         worker_path = datmo_deploy['deploy']['celery_services']['worker_path']
                         config_method_names = datmo_deploy['deploy']['celery_services']['methods']