From 31bef5af8bd0244fd0bf19ea84326530ef7e6390 Mon Sep 17 00:00:00 2001
From: huchukato <huchukato@gmail.com>
Date: Fri, 27 Feb 2026 09:21:44 +0100
Subject: [PATCH] Fix dummy module classes - make Logger, BuilderFlag, etc.
 callable instead of ModuleType

---
 .DS_Store        | Bin 6148 -> 6148 bytes
 trt_utilities.py |  49 ++++++++++++++++++++++++++++++++++-------------
 2 files changed, 36 insertions(+), 13 deletions(-)

diff --git a/.DS_Store b/.DS_Store
index 545b76e747ae265aca5f22bfb0c11eb6f1cefc49..15206a5a95d17c9278f52e143261bbde03fed9d6 100644
GIT binary patch
delta 106
zcmZoMXffEJ#uU5d2Ll5G3xgg*IzuKyNp8N2OHxjL5>SjI<5x|=krl@sQRP$c$`@o9
U1}Ep|76A1CP57|6nW<3}0A&##Hvj+t

delta 106
zcmZoMXffEJ#uRJ&hk=2Cg+Y%YogtH<Bsbs1B`GIA2`I)9^W<{bv=zr4QRP$c$`@o9
V1}Ep|76A1yFtBgf+|1M{3IIJ!9i0FG

diff --git a/trt_utilities.py b/trt_utilities.py
index d90b5ea..830d97d 100644
--- a/trt_utilities.py
+++ b/trt_utilities.py
@@ -35,19 +35,42 @@ def get_trt():
             # Create dummy tensorrt module for graceful fallback
             import types
             trt = types.ModuleType('tensorrt')
-            trt.Logger = types.ModuleType('Logger')
-            trt.Logger.ERROR = 0
-            trt.Logger.WARNING = 1
-            trt.BuilderFlag = types.ModuleType('BuilderFlag')
-            trt.BuilderFlag.FP16 = 1
-            trt.BuilderFlag.REFIT = 2
-            trt.OnnxParserFlag = types.ModuleType('OnnxParserFlag')
-            trt.OnnxParserFlag.NATIVE_INSTANCENORM = 1
-            trt.TensorIOMode = types.ModuleType('TensorIOMode')
-            trt.TensorIOMode.INPUT = 0
-            trt.TensorIOMode.OUTPUT = 1
-            trt.nptype = lambda x: np.float32  # dummy fallback
-            trt.IProgressMonitor = object  # dummy fallback
+            
+            # Create Logger class with ERROR attribute
+            class DummyLogger:
+                def __init__(self, level):
+                    self.level = level
+                ERROR = 0
+                WARNING = 1
+            
+            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
+            trt.IProgressMonitor = object
+            
             _trt = trt
             _trt_available = False
     return _trt