[Python] Use IL-specific types in return type annotations for get_basic_block_at

Author: tbodt

python/highlevelil.py

@@ -2647,7 +2647,7 @@ def _instantiate_block(self, handle):
 	def basic_blocks(self) -> 'function.HighLevelILBasicBlockList':
 		return function.HighLevelILBasicBlockList(self)
 
-	def get_basic_block_at(self, index: int) -> Optional['basicblock.BasicBlock']:
+	def get_basic_block_at(self, index: int) -> Optional['HighLevelILBasicBlock']:
 		"""
 		``get_basic_block_at`` returns the BasicBlock at the given HLIL instruction ``index``.
 

python/lowlevelil.py

@@ -3410,7 +3410,7 @@ def _instantiate_block(self, handle):
 	def basic_blocks(self) -> 'function.LowLevelILBasicBlockList':
 		return function.LowLevelILBasicBlockList(self)
 
-	def get_basic_block_at(self, index: int) -> Optional['basicblock.BasicBlock']:
+	def get_basic_block_at(self, index: int) -> Optional['LowLevelILBasicBlock']:
 		"""
 		``get_basic_block_at`` returns the BasicBlock at the given LLIL instruction ``index``.
 

python/mediumlevelil.py

@@ -3429,7 +3429,7 @@ def _instantiate_block(self, handle):
 	def basic_blocks(self) -> 'function.MediumLevelILBasicBlockList':
 		return function.MediumLevelILBasicBlockList(self)
 
-	def get_basic_block_at(self, index: int) -> Optional['basicblock.BasicBlock']:
+	def get_basic_block_at(self, index: int) -> Optional['MediumLevelILBasicBlock']:
 		"""
 		``get_basic_block_at`` returns the BasicBlock at the given MLIL instruction ``index``.