From c665f48f3bc0ad68a8477b748d23effa98baa6ca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Kuligowski?= <mkuligowski@habana.ai>
Date: Fri, 25 Oct 2024 13:36:27 +0200
Subject: [PATCH] Update collect_env.py

---
 collect_env.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/collect_env.py b/collect_env.py
index 443be171cd93..055c57f8f84d 100644
--- a/collect_env.py
+++ b/collect_env.py
@@ -254,8 +254,8 @@ def get_nvidia_smi():
 
 
 def get_hpu_info():
-    command = ["hl-smi", "-q", "-d", "PRODUCT"]
     try:
+        command = ["hl-smi", "-q", "-d", "PRODUCT"]
         lines = subprocess.Popen(command, stdout=subprocess.PIPE, universal_newlines=True).stdout.readlines()
         lines = [l.strip('\t') for l in lines]
         hpu_count = None
@@ -283,7 +283,7 @@ def get_hpu_info():
         if hpu_model is None:
             return ('N/A', hpu_driver)
         return (f'{hpu_count}x {hpu_model}', hpu_driver)
-    except Exception as e:
+    except:
         return ('N/A', 'N/A')