From 23f3cbcea718e322224f2aee00964da29ffd796c Mon Sep 17 00:00:00 2001
From: varun-r-mallya <varunrmallya@gmail.com>
Date: Thu, 2 Oct 2025 01:43:05 +0530
Subject: [PATCH] add type annotations

---
 pythonbpf/helper/__init__.py     | 2 +-
 pythonbpf/helper/helper_utils.py | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/pythonbpf/helper/__init__.py b/pythonbpf/helper/__init__.py
index 1728eb8..17f3198 100644
--- a/pythonbpf/helper/__init__.py
+++ b/pythonbpf/helper/__init__.py
@@ -1,4 +1,4 @@
 from .helper_utils import HelperHandlerRegistry
 from .bpf_helper_handler import handle_helper_call
 
-__all__ = [HelperHandlerRegistry, handle_helper_call]
+__all__ = ["HelperHandlerRegistry", "handle_helper_call"]
diff --git a/pythonbpf/helper/helper_utils.py b/pythonbpf/helper/helper_utils.py
index ed3791b..8e658f0 100644
--- a/pythonbpf/helper/helper_utils.py
+++ b/pythonbpf/helper/helper_utils.py
@@ -1,5 +1,7 @@
 import ast
 import logging
+from collections.abc import Callable
+
 from llvmlite import ir
 from pythonbpf.expr_pass import eval_expr
 
@@ -9,7 +11,7 @@ logger = logging.getLogger(__name__)
 class HelperHandlerRegistry:
     """Registry for BPF helpers"""
 
-    _handlers = {}
+    _handlers: dict[str, Callable] = {}
 
     @classmethod
     def register(cls, helper_name):