From 98c8e3fa373af2be415f14c2b536343e40cf88e3 Mon Sep 17 00:00:00 2001 From: Tyler Gu Date: Sun, 25 Feb 2024 19:45:18 -0600 Subject: [PATCH] Fix CR loading Signed-off-by: Tyler Gu --- acto/post_process/post_diff_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/acto/post_process/post_diff_test.py b/acto/post_process/post_diff_test.py index a1c81687cf..33f84e96e4 100644 --- a/acto/post_process/post_diff_test.py +++ b/acto/post_process/post_diff_test.py @@ -18,6 +18,7 @@ import pandas as pd import pydantic +import yaml from deepdiff import DeepDiff from deepdiff.helper import CannotCompare from deepdiff.model import DiffLevel, TreeResult @@ -710,7 +711,7 @@ def check(self, workdir: str, num_workers: int = 1): trial_dirs = glob.glob(os.path.join(workdir, "trial-*")) with open(self.config.seed_custom_resource, "r", encoding="utf-8") as f: - seed_cr = json.load(f) + seed_cr = yaml.load(f, Loader=yaml.FullLoader) seed_input_digest = hashlib.md5( json.dumps(seed_cr, sort_keys=True).encode("utf-8") ).hexdigest()