fix code style

This commit is contained in:
Yacine Elhamer
2023-06-03 01:12:56 +01:00
parent 5b903ca4f3
commit be5ada26ea
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -92,7 +92,7 @@ class Shdr:
buf: bytes
@classmethod
def from_viv(cls, section, buf: bytes):
def from_viv(cls, section, buf: bytes) -> "Shdr":
return cls(
section.sh_name,
section.sh_type,
+1 -1
View File
@@ -51,7 +51,7 @@ class VivisectFeatureExtractor(FeatureExtractor):
def get_functions(self) -> Iterator[FunctionHandle]:
cache = {}
for va in sorted(self.vw.getFunctions()):
yield FunctionHandle(address=AbsoluteVirtualAddress(va), inner=viv_utils.Function(self.vw, va), ctx={"cache": cache})
yield FunctionHandle(address=AbsoluteVirtualAddress(va), inner=viv_utils.Function(self.vw, va), ctx={"cache":cache})
def extract_function_features(self, fh: FunctionHandle) -> Iterator[Tuple[Feature, Address]]:
yield from capa.features.extractors.viv.function.extract_features(fh)