From 762dc0b821cf50635e44aac2676be28b58b29ee7 Mon Sep 17 00:00:00 2001 From: Arjun Suresh Date: Wed, 14 Aug 2024 23:22:28 +0100 Subject: [PATCH] Remove pandas deps for app-mlperf-inference --- script/app-mlperf-inference/_cm.yaml | 5 +++++ script/app-mlperf-inference/customize.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/script/app-mlperf-inference/_cm.yaml b/script/app-mlperf-inference/_cm.yaml index d81eab650..6d44c6066 100644 --- a/script/app-mlperf-inference/_cm.yaml +++ b/script/app-mlperf-inference/_cm.yaml @@ -104,6 +104,11 @@ deps: - tags: get,mlperf,inference,utils + - tags: install,pip-package,for-cmind-python,_package.pandas + enable_if_env: + CM_PROFILE_NVIDIA_POWER: + - on + posthook_deps: - tags: get,mlperf,sut,description #populate system meta information like framework diff --git a/script/app-mlperf-inference/customize.py b/script/app-mlperf-inference/customize.py index f134735e8..f478c6d67 100644 --- a/script/app-mlperf-inference/customize.py +++ b/script/app-mlperf-inference/customize.py @@ -9,7 +9,6 @@ import platform import sys import mlperf_utils -import pandas as pd import re from datetime import datetime,timezone @@ -449,6 +448,7 @@ def postprocess(i): # portion of the code where the avg utilisation and system informations are extracted # NOTE: The section is under development and print statements are added for further debugging if env.get('CM_PROFILE_NVIDIA_POWER', '') == "on": + import pandas as pd system_utilisation_info_dump = {} logs_dir = output_dir # logs_dir = env.get('CM_LOGS_DIR', env['CM_RUN_DIR'])