From de3e66852a8a35b02da7f1be207fca70db13c567 Mon Sep 17 00:00:00 2001
From: Bobby Wang <wbo4958@gmail.com>
Date: Thu, 6 Nov 2025 16:12:53 +0800
Subject: [PATCH] Fix listener unregister issue

It's a regression issue. The Python listener should be set to None when it fails to register.

Signed-off-by: Bobby Wang <wbo4958@gmail.com>
---
 nds/PysparkBenchReport.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nds/PysparkBenchReport.py b/nds/PysparkBenchReport.py
index 6f56c27..1dcf65c 100644
--- a/nds/PysparkBenchReport.py
+++ b/nds/PysparkBenchReport.py
@@ -76,6 +76,7 @@ def _register_python_listener(self):
             listener.register()
         except TypeError as e:
             print("Not found com.nvidia.spark.rapids.listener.Manager", str(e))
+            listener = None
         return listener
 
     def _get_spark_conf(self):