From 6bccbf3425e2e336e654faf220df5d59ab7162e0 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Mon, 18 Dec 2023 11:45:33 -0500 Subject: [PATCH] Use str rep of the OSError --- userbenchmark/test_bench/run.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/userbenchmark/test_bench/run.py b/userbenchmark/test_bench/run.py index cbca58099f..f27c81e4a1 100644 --- a/userbenchmark/test_bench/run.py +++ b/userbenchmark/test_bench/run.py @@ -125,7 +125,7 @@ def run_config(config: TorchBenchModelConfig, metrics: List[str], dryrun: bool=F print(" [oserror]", flush=True) result = {} for metric in metrics: - result[metric] = e + result[metric] = str(e) return result def run_config_accuracy(config: TorchBenchModelConfig, metrics: List[str], dryrun: bool=False) -> Dict[str, str]: @@ -143,7 +143,7 @@ def run_config_accuracy(config: TorchBenchModelConfig, metrics: List[str], dryru return {"accuracy": "not_implemented"} except OSError as e: print(" [oserror]", flush=True) - return {"accuracy": f"{e.__str__}"} + return {"accuracy": str(e)} def parse_known_args(args): parser = argparse.ArgumentParser()