fix wrong type hint

scripts/Infrastructure/AllocCall.py

@@ -25,7 +25,7 @@ alloc_template = """
 
 class AllocCall(Instruction):
     @override
-    def __init__(self, type: str, num_elements: int, name: str = "buf", use_malloc: bool = False):
+    def __init__(self, type: str, num_elements: str, name: str = "buf", use_malloc: bool = False):
         """
         Creates a New allocation Call
 
@@ -62,7 +62,7 @@ class AllocCall(Instruction):
 
         return s
 
-    def set_num_elements(self, num_elements: int):
+    def set_num_elements(self, num_elements: str):
         self._num_elements = num_elements
 
     def set_name(self, name: str):
@@ -77,7 +77,7 @@ class AllocCall(Instruction):
     def set_use_calloc(self):
         self._use_malloc = False
 
-    def get_num_elements(self) -> int:
+    def get_num_elements(self) -> str:
         return self._num_elements
 
     def get_name(self) -> str: