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):