Skip to content

Commit

Permalink
ndarray
Browse files Browse the repository at this point in the history
  • Loading branch information
tsmonteiro committed Aug 27, 2024
1 parent 64fff8e commit a2020c0
Showing 1 changed file with 21 additions and 1 deletion.
22 changes: 21 additions & 1 deletion runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit a2020c0

Please sign in to comment.