From 2c90f98639823f030ee26be287786b1c6c752851 Mon Sep 17 00:00:00 2001
From: Amit Pant <50907565+amitpant7@users.noreply.github.com>
Date: Thu, 14 Dec 2023 10:29:35 +0545
Subject: [PATCH] Update latency_table.py

in yam.load(), it now expects a loader function also, otherwise the file won't open.
---
 ofa/tutorial/latency_table.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ofa/tutorial/latency_table.py b/ofa/tutorial/latency_table.py
index 7543641..c304e69 100644
--- a/ofa/tutorial/latency_table.py
+++ b/ofa/tutorial/latency_table.py
@@ -14,7 +14,7 @@ def __init__(
             fname = url
 
         with open(fname, "r") as fp:
-            self.lut = yaml.load(fp)
+            self.lut = yaml.load(fp, Loader = yaml.FullLoader))
 
     @staticmethod
     def repr_shape(shape):