From a2020c049f43447e51fadde6e9076e1542a19dd2 Mon Sep 17 00:00:00 2001 From: Thiago Date: Tue, 27 Aug 2024 11:31:39 -0300 Subject: [PATCH] ndarray --- runner.py | 22 +++++++++++++++++++++- 1 file changed, 21 insertions(+), 1 deletion(-) diff --git a/runner.py b/runner.py index 527a7cb..9f3d470 100644 --- a/runner.py +++ b/runner.py @@ -258,7 +258,7 @@ def run_with_params(params, mode="cli"): else: if isinstance(resultDict, dict) or isinstance(resultDict, list): with open('test_results.json', 'w', encoding='utf-8') as f: - json.dump(resultDict, f, ensure_ascii=False, indent=4) + json.dump(fixDictTypes(resultDict), f, ensure_ascii=False, indent=4) else: with open('test_results.json', "w") as f: for line in resultDict: @@ -307,6 +307,26 @@ def run_with_params(params, mode="cli"): if mode == "operator": return statusList +import numpy as np +from typing import cast +def fixDictTypes(dictObj): + if isinstance(dictObj, list): + for o in dictObj.items(): + for k, v in o: + if isinstance(v, dict): + fixDictTypes(v) + elif(isinstance(v, np.ndarray)): + v = cast( np.array, v).tolist() + else: + for k, v in dictObj.items(): + if isinstance(v, dict): + fixDictTypes(v) + elif(isinstance(v, np.ndarray)): + v = cast( np.array, v).tolist() + + return dictObj + + def tercenBool(value): if value == "true": return True