remove tbaa_gen and make IR generator module

varun-r-mallya · varun-r-mallya · commit d3f0e3b2ef28 · 2025-10-14T03:09:18.000+05:30
diff --git a/pythonbpf/tbaa_gen/__init__.py b/pythonbpf/tbaa_gen/__init__.py
diff --git a/pythonbpf/vmlinux_parser/import_detector.py b/pythonbpf/vmlinux_parser/import_detector.py
@@ -5,7 +5,7 @@
 import inspect
 
 from .dependency_handler import DependencyHandler
-from .ir_generation import IRGenerator
+from .ir_gen import IRGenerator
 from .class_handler import process_vmlinux_class
 
 logger = logging.getLogger(__name__)
diff --git a/pythonbpf/vmlinux_parser/ir_gen/__init__.py b/pythonbpf/vmlinux_parser/ir_gen/__init__.py
@@ -0,0 +1,3 @@
+from .ir_generation import IRGenerator
+
+__all__ = ["IRGenerator"]
diff --git a/pythonbpf/vmlinux_parser/ir_gen/ir_generation.py b/pythonbpf/vmlinux_parser/ir_gen/ir_generation.py
@@ -1,5 +1,5 @@
 import logging
-from .dependency_handler import DependencyHandler
+from pythonbpf.vmlinux_parser.dependency_handler import DependencyHandler
 
 logger = logging.getLogger(__name__)
 

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+from .ir_generation import IRGenerator`
	`2`	`+`
	`3`	`+__all__ = ["IRGenerator"]`