Update DB module for new sysinfo keys

Signed-off-by: colramos-amd <colramos@amd.com>
Esse commit está contido em:
colramos-amd
2024-03-12 12:24:49 -05:00
commit 706d3bb9f0
+7 -2
Ver Arquivo
@@ -66,8 +66,13 @@ class DatabaseConnector:
sys_info = os.path.join(self.connection_info["workload"], "sysinfo.csv")
if os.path.isfile(sys_info):
sys_info = pd.read_csv(sys_info)
soc = sys_info["name"][0]
name = sys_info["workload_name"][0]
try:
soc = sys_info["gpu_model"][0]
name = sys_info["workload_name"][0]
except KeyError as e:
console_error(
f"Outdated workload. Cannot find {e} field. Please reprofile to update."
)
else:
console_error(
"database", "Unable to parse SoC and/or workload name from sysinfo.csv"