From 3a427795eac910a3f68cd1e07a3e997e152d49cf Mon Sep 17 00:00:00 2001
From: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
Date: Thu, 7 Sep 2023 15:37:06 +0000
Subject: [PATCH] metrics: Use TensorFlow optimized image

This PR replaces the ubuntu image for one which has TensorFlow optimized
for kata metrics.

Fixes #7866

Signed-off-by: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
---
 .../machine_learning/resnet50_fp32_dockerfile/Dockerfile        | 2 +-
 .../machine_learning/resnet50_int8_dockerfile/Dockerfile        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile b/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile
index a0e7964560..edbdf7370a 100644
--- a/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile
+++ b/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile
@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 # Usage: FROM [image name]
-FROM ubuntu:20.04
+FROM intel/intel-optimized-tensorflow:2.9.1
 
 ENV DEBIAN_FRONTEND=noninteractive
 
diff --git a/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile b/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile
index a033475ee5..6e0c1e6119 100644
--- a/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile
+++ b/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile
@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 # Usage: FROM [image name]
-FROM ubuntu:20.04
+FROM intel/intel-optimized-tensorflow:2.9.1
 
 ENV DEBIAN_FRONTEND=noninteractive