commit 81770986ac19d75c8b2ffebc87f97aaa81ad714f
author Richard Burton <richard.burton@arm.com> Wed Apr 29 20:07:09 2020 +0100
committer Richard <richard.burton@arm.com> Tue May 05 12:34:15 2020 +0000
tree 1b41d822f0ba62a25c442ca8ef98e2a2aaccdca6
parent 884746d2677e50c47b4bc70a2829a777344f73c4

Changing print statement to logging

Signed-off-by: Richard Burton <richard.burton@arm.com>
Change-Id: I38a929ec6ba8c6df0a84122a3329d80f86b5c0d0

python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py

diff --git a/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py b/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py
index b33356c..22b8768 100644
--- a/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py
+++ b/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py
@@ -5,6 +5,7 @@
 WorkloadTensors are the inputTensors and outputTensors that are consumed by IRuntime.EnqueueWorkload.
 """
 from typing import Union, List, Tuple
+import logging
 
 import numpy as np
 
@@ -121,6 +122,6 @@
     arrays = []
     for index, (_, tensor) in enumerate(workload_tensors):
         arrays.append(tensor.get_memory_area().reshape(list(tensor.GetShape())))
-        print("Workload tensor {} shape: {}".format(index, tensor.GetShape()))
+        logging.info("Workload tensor {} shape: {}".format(index, tensor.GetShape()))
 
     return arrays