From 57e33fb16e1db84323fb3500f70332f9e66a44c8 Mon Sep 17 00:00:00 2001
From: Abhiram V <61599526+Anon-Artist@users.noreply.github.com>
Date: Tue, 19 Jan 2021 00:16:46 +0530
Subject: [PATCH] Update test.py with yaml.SafeLoader (#1969)

---
 test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.py b/test.py
index 36d18132c782..1c6ea103489b 100644
--- a/test.py
+++ b/test.py
@@ -68,7 +68,7 @@ def test(data,
     model.eval()
     is_coco = data.endswith('coco.yaml')  # is COCO dataset
     with open(data) as f:
-        data = yaml.load(f, Loader=yaml.FullLoader)  # model dict
+        data = yaml.load(f, Loader=yaml.SafeLoader)  # model dict
     check_dataset(data)  # check
     nc = 1 if single_cls else int(data['nc'])  # number of classes
     iouv = torch.linspace(0.5, 0.95, 10).to(device)  # iou vector for mAP@0.5:0.95