fix init

Signed-off-by: Justin Chu <[email protected]>

Author: justinchuby

src/onnx_ir/__init__.py

@@ -149,17 +149,15 @@
 )
 from onnx_ir.serde import TensorProtoTensor, from_onnx_text, from_proto, to_proto
 
-__version__ = "0.1.0"
 DEBUG = False
 
 
 def __set_module() -> None:
     """Set the module of all functions in this module to this public module."""
     global_dict = globals()
     for name in __all__:
-        if hasattr(global_dict[name], "__module__"):
-            # Set the module of the function to this module
-            global_dict[name].__module__ = __name__
+        global_dict[name].__module__ = __name__
 
 
 __set_module()
+__version__ = "0.1.0"