From 74f2cdb9005c507626646878e07c898780c552e3 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 26 Oct 2022 17:12:36 +0100 Subject: [PATCH] =?UTF-8?q?train=5Fpredict:=20.list()=20=E2=86=92=20.tolis?= =?UTF-8?q?t()?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- aimodel/src/subcommands/train_predict.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/aimodel/src/subcommands/train_predict.py b/aimodel/src/subcommands/train_predict.py index ac84492..c96ce15 100644 --- a/aimodel/src/subcommands/train_predict.py +++ b/aimodel/src/subcommands/train_predict.py @@ -165,11 +165,11 @@ def do_jsonl(args, ai, dataset, model_params): item_obj = {} if "rainfall_actual" in args.log: - item_obj["rainfall_actual"] = rainfall_actual_batch[i_batch].numpy().list() + item_obj["rainfall_actual"] = rainfall_actual_batch[i_batch].numpy().tolist() if "water_actual" in args.log: - item_obj["water_actual"] = water_actual.numpy().list() + item_obj["water_actual"] = water_actual.numpy().tolist() if "water_predict" in args.log: - item_obj["water_predict"] = water_predict.numpy().list() + item_obj["water_predict"] = water_predict.numpy().tolist() handle.write(json.dumps(item_obj, separators=(',', ':'))+"\n") # Ref https://stackoverflow.com/a/64710892/1460422