Changing print statement to logging

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

diff --git a/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py b/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py
index b33356c..22b8768 100644 (file)
--- a/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py
+++ b/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py
@@ -5,6 +5,7 @@ This file contains functions relating to WorkloadTensors.
 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 @@ def workload_tensors_to_ndarray(workload_tensors: List[Tuple[int, Union[Tensor,
     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