From d3f0e3b2ef2881d89718b38e178335a569d5b506 Mon Sep 17 00:00:00 2001
From: varun-r-mallya <varunrmallya@gmail.com>
Date: Tue, 14 Oct 2025 03:09:18 +0530
Subject: [PATCH] remove tbaa_gen and make IR generator module

---
 pythonbpf/tbaa_gen/__init__.py                         | 0
 pythonbpf/vmlinux_parser/import_detector.py            | 2 +-
 pythonbpf/vmlinux_parser/ir_gen/__init__.py            | 3 +++
 pythonbpf/vmlinux_parser/{ => ir_gen}/ir_generation.py | 2 +-
 4 files changed, 5 insertions(+), 2 deletions(-)
 delete mode 100644 pythonbpf/tbaa_gen/__init__.py
 create mode 100644 pythonbpf/vmlinux_parser/ir_gen/__init__.py
 rename pythonbpf/vmlinux_parser/{ => ir_gen}/ir_generation.py (85%)

diff --git a/pythonbpf/tbaa_gen/__init__.py b/pythonbpf/tbaa_gen/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/pythonbpf/vmlinux_parser/import_detector.py b/pythonbpf/vmlinux_parser/import_detector.py
index 2ce9cb5..f5789ce 100644
--- a/pythonbpf/vmlinux_parser/import_detector.py
+++ b/pythonbpf/vmlinux_parser/import_detector.py
@@ -5,7 +5,7 @@ import importlib
 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
new file mode 100644
index 0000000..3a13651
--- /dev/null
+++ 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_generation.py b/pythonbpf/vmlinux_parser/ir_gen/ir_generation.py
similarity index 85%
rename from pythonbpf/vmlinux_parser/ir_generation.py
rename to pythonbpf/vmlinux_parser/ir_gen/ir_generation.py
index c66ba11..e4dae7d 100644
--- a/pythonbpf/vmlinux_parser/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__)