From facd447f95f8cf07524a6d43f19a41344df8f680 Mon Sep 17 00:00:00 2001
From: huchukato <huchukato@gmail.com>
Date: Fri, 27 Feb 2026 09:54:50 +0100
Subject: [PATCH] Fix DummyIProgressMonitor scope issue - move dummy classes to
 global scope

---
 trt_utilities.py | 65 ++++++++++++++++++++++--------------------------
 1 file changed, 30 insertions(+), 35 deletions(-)

diff --git a/trt_utilities.py b/trt_utilities.py
index aee6b91..daf9b16 100644
--- a/trt_utilities.py
+++ b/trt_utilities.py
@@ -23,6 +23,34 @@ import sys
 _trt = None
 _trt_available = False
 
+# Create dummy classes that need to be accessible globally
+class DummyIProgressMonitor:
+    def __init__(self):
+        pass
+    def phase_start(self, *args, **kwargs):
+        pass
+    def phase_finish(self, *args, **kwargs):
+        pass
+    def step_complete(self, *args, **kwargs):
+        return True
+
+class DummyLogger:
+    def __init__(self, level):
+        self.level = level
+    ERROR = 0
+    WARNING = 1
+
+class DummyBuilderFlag:
+    FP16 = 1
+    REFIT = 2
+
+class DummyOnnxParserFlag:
+    NATIVE_INSTANCENORM = 1
+
+class DummyTensorIOMode:
+    INPUT = 0
+    OUTPUT = 1
+
 def get_trt():
     global _trt, _trt_available
     if _trt is None:
@@ -36,49 +64,16 @@ def get_trt():
             import types
             trt = types.ModuleType('tensorrt')
             
-            # Create Logger class with ERROR attribute
-            class DummyLogger:
-                def __init__(self, level):
-                    self.level = level
-                ERROR = 0
-                WARNING = 1
-            
+            # Use pre-defined dummy classes
             trt.Logger = DummyLogger
-            
-            # Create BuilderFlag class with flags
-            class DummyBuilderFlag:
-                FP16 = 1
-                REFIT = 2
-            
             trt.BuilderFlag = DummyBuilderFlag
-            
-            # Create OnnxParserFlag class with flags
-            class DummyOnnxParserFlag:
-                NATIVE_INSTANCENORM = 1
-            
             trt.OnnxParserFlag = DummyOnnxParserFlag
-            
-            # Create TensorIOMode class with modes
-            class DummyTensorIOMode:
-                INPUT = 0
-                OUTPUT = 1
-            
             trt.TensorIOMode = DummyTensorIOMode
             
             # Create dummy nptype function
             trt.nptype = lambda x: np.float32
             
-            # Create dummy IProgressMonitor class that works without inheritance
-            class DummyIProgressMonitor:
-                def __init__(self):
-                    pass
-                def phase_start(self, *args, **kwargs):
-                    pass
-                def phase_finish(self, *args, **kwargs):
-                    pass
-                def step_complete(self, *args, **kwargs):
-                    return True
-            
+            # Use pre-defined dummy class
             trt.IProgressMonitor = DummyIProgressMonitor
             
             _trt = trt